Fix regression with DatasetNotFoundError instead of GatedRepoError

huggingface · Sep 26, 2024 · ad74707 · ad74707
1 parent 22de57f
commit ad74707
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/services/worker/tests/job_runners/config/test_split_names.py b/services/worker/tests/job_runners/config/test_split_names.py
@@ -173,7 +173,7 @@ def test_doesnotexist(
         # should we really test the following cases?
         # The assumption is that the dataset exists and is accessible with the token
         ("does_not_exist", False, "SplitNamesFromStreamingError", "DatasetNotFoundError"),
-        ("gated", False, "SplitNamesFromStreamingError", "GatedRepoError"),  # See: huggingface/datasets#7109
+        ("gated", False, "SplitNamesFromStreamingError", "DatasetNotFoundError"),
         ("private", False, "SplitNamesFromStreamingError", "DatasetNotFoundError"),
     ],
 )

diff --git a/services/worker/tests/job_runners/dataset/test_config_names.py b/services/worker/tests/job_runners/dataset/test_config_names.py
@@ -104,7 +104,7 @@ def test_compute_too_many_configs(
         # should we really test the following cases?
         # The assumption is that the dataset exists and is accessible with the token
         ("does_not_exist", False, "RetryableConfigNamesError", "DatasetNotFoundError"),
-        ("gated", False, "RetryableConfigNamesError", "GatedRepoError"),  # See: huggingface/datasets#7109
+        ("gated", False, "RetryableConfigNamesError", "DatasetNotFoundError"),
         ("private", False, "RetryableConfigNamesError", "DatasetNotFoundError"),
     ],
 )