dsmueller commited on
Commit
bfcd978
1 Parent(s): da96b77

Added data_dir

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  if __name__ == "__main__":
8
  dataset_name="ai-aerospace/ac-text-embedding-ada-002-ams-test"
9
  # Load dataset using datasets.load_dataset()
10
- dataset = datasets.load_dataset(dataset_name, split="train")
11
  print("Dataset loaded using datasets.load_dataset().")
12
 
13
  df = dataset.to_pandas()
 
7
  if __name__ == "__main__":
8
  dataset_name="ai-aerospace/ac-text-embedding-ada-002-ams-test"
9
  # Load dataset using datasets.load_dataset()
10
+ dataset = datasets.load_dataset(dataset_name, split="train",data_dir='/code')
11
  print("Dataset loaded using datasets.load_dataset().")
12
 
13
  df = dataset.to_pandas()