Fix patch in tests

huggingface · Aug 27, 2024 · 67f9b68 · 67f9b68
1 parent 5c1d27e
commit 67f9b68
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py
@@ -164,9 +164,9 @@ def test_revision_exists(self):
         assert not self._api.revision_exists(self.repo_id, "main", token=False)  # private repo
         assert not self._api.revision_exists("repo-that-does-not-exist", "main")  # missing repo
 
-    @patch("huggingface_hub.file_download.ENDPOINT", "https://hub-ci.huggingface.co")
+    @patch("huggingface_hub.constants.ENDPOINT", "https://hub-ci.huggingface.co")
     @patch(
-        "huggingface_hub.file_download.HUGGINGFACE_CO_URL_TEMPLATE",
+        "huggingface_hub.constants.HUGGINGFACE_CO_URL_TEMPLATE",
         "https://hub-ci.huggingface.co/{repo_id}/resolve/{revision}/{filename}",
     )
     def test_file_exists(self):