Test loading GitHub dataset from Hub

huggingface · Mar 30, 2022 · fae366b · fae366b
1 parent a4c9be6
commit fae366b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_load.py b/tests/test_load.py
@@ -23,7 +23,6 @@
 from datasets.load import (
     CachedDatasetModuleFactory,
     CachedMetricModuleFactory,
-    GithubDatasetModuleFactory,
     GithubMetricModuleFactory,
     HubDatasetModuleFactoryWithoutScript,
     HubDatasetModuleFactoryWithScript,
@@ -183,9 +182,9 @@ def setUp(self):
             hf_modules_cache=self.hf_modules_cache,
         )
 
-    def test_GithubDatasetModuleFactory(self):
+    def test_HubDatasetModuleFactoryWithScript_with_github_dataset(self):
         # "wmt_t2t" has additional imports (internal)
-        factory = GithubDatasetModuleFactory(
+        factory = HubDatasetModuleFactoryWithScript(
             "wmt_t2t", download_config=self.download_config, dynamic_modules_path=self.dynamic_modules_path
         )
         module_factory_result = factory.get_module()