fix broken zip directory approach

GRAAL-Research · May 18, 2023 · 09b9ef4 · 09b9ef4
1 parent 1c877c1
commit 09b9ef4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/RISCBAC.py b/RISCBAC.py
@@ -92,7 +92,7 @@ def _split_generators(self, dl_manager):
                 datasets.SplitGenerator(
                     name="full_en",
                     gen_kwargs={
-                        "filepath": os.path.join(data_dir, "en", "en.jsonl"),
+                        "filepath": os.path.join(data_dir, "en.jsonl"),
                     },
                 )
             ]
@@ -101,7 +101,7 @@ def _split_generators(self, dl_manager):
                 datasets.SplitGenerator(
                     name="full_fr",
                     gen_kwargs={
-                        "filepath": os.path.join(data_dir, "fr", "fr.jsonl"),
+                        "filepath": os.path.join(data_dir, "fr.jsonl"),
                     },
                 ),
             ]