Merge pull request #93 from OpenGPTX/fix-offline

Fix caching
OpenGPTX · Oct 20, 2023 · c6579ae · c6579ae
2 parents 93e3595 + 94b8230
commit c6579ae
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lm_eval/tasks/opengptx/germeval2018.py b/lm_eval/tasks/opengptx/germeval2018.py
@@ -100,6 +100,17 @@ class GermEval2018(Task):
         "OFFENSE": "Beleidigung",
     }
 
+    def download(self, data_dir=None, cache_dir=None, download_mode=None):
+        args = ()
+        kwargs = dict(
+            path=self.DATASET_PATH,
+            name=self.DATASET_NAME,
+            data_dir=data_dir,
+            cache_dir=cache_dir,
+            download_mode=download_mode,
+        )
+        self._download_pushed(args, kwargs, data_dir, cache_dir, download_mode)
+
     def has_training_docs(self):
         return True