sps44 commited on
Commit
18df44d
1 Parent(s): 3afaf74

fix parquet path

Browse files
Files changed (1) hide show
  1. prepare.py +1 -1
prepare.py CHANGED
@@ -22,7 +22,7 @@ if __name__ == "__main__":
22
  cache_file = "dataset_cache.parquet"
23
  if os.path.exists(cache_file):
24
  # Load dataset from cache
25
- df = pd.read_parquet('cifar100-enriched-cv.parquet')
26
  print("Dataset loaded from cache.")
27
  else:
28
  # Load dataset using datasets.load_dataset()
 
22
  cache_file = "dataset_cache.parquet"
23
  if os.path.exists(cache_file):
24
  # Load dataset from cache
25
+ df = pd.read_parquet(cache_file)
26
  print("Dataset loaded from cache.")
27
  else:
28
  # Load dataset using datasets.load_dataset()