Fix missing default data_dir during splits discovery

qcri · Sep 18, 2023 · 0adaa49 · 0adaa49
1 parent e7d52db
commit 0adaa49
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/llmebench/utils.py b/llmebench/utils.py
@@ -51,6 +51,7 @@ def get_data_paths(config, split):
     assert split in ["train", "test"]
 
     dataset_args = config.get("dataset_args", {})
+    dataset_args["data_dir"] = ""
     dataset = config["dataset"](**dataset_args)
 
     if split == "test":