diff --git a/tests/utils/test_hub_utils.py b/tests/utils/test_hub_utils.py index 89e41e4a0738..5ff1ed679e6e 100644 --- a/tests/utils/test_hub_utils.py +++ b/tests/utils/test_hub_utils.py @@ -128,14 +128,14 @@ def test_get_file_from_repo_local(self): self.assertIsNone(get_file_from_repo(tmp_dir, "b.txt")) - @unittest.skip("Test is broken, fix me Wauplain!") def test_get_file_gated_repo(self): """Test download file from a gated repo fails with correct message when not authenticated.""" with self.assertRaisesRegex(EnvironmentError, "You are trying to access a gated repo."): - cached_file(GATED_REPO, README_FILE, use_auth_token=False) + # All files except README.md are protected on a gated repo. + cached_file(GATED_REPO, "gated_file.txt", use_auth_token=False) - @unittest.skip("Test is broken, fix me Wauplain!") def test_has_file_gated_repo(self): """Test check file existence from a gated repo fails with correct message when not authenticated.""" with self.assertRaisesRegex(EnvironmentError, "is a gated repository"): - has_file(GATED_REPO, README_FILE, use_auth_token=False) + # All files except README.md are protected on a gated repo. + has_file(GATED_REPO, "gated_file.txt", use_auth_token=False)