Fix test_list_spaces_linked (#2707)

huggingface · Dec 11, 2024 · 51b866f · 51b866f
1 parent e2bc481
commit 51b866f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py
@@ -2318,13 +2318,13 @@ def test_list_spaces_with_datasets(self):
         assert "wikipedia" in spaces[0].datasets
 
     def test_list_spaces_linked(self):
-        space_id = "open-llm-leaderboard/open_llm_leaderboard"
+        space_id = "stabilityai/stable-diffusion"
 
-        spaces = list(self._api.list_spaces(search=space_id))
+        spaces = [space for space in self._api.list_spaces(search=space_id) if space.id == space_id]
         assert spaces[0].models is None
         assert spaces[0].datasets is None
 
-        spaces = list(self._api.list_spaces(search=space_id, linked=True))
+        spaces = [space for space in self._api.list_spaces(search=space_id, linked=True) if space.id == space_id]
         assert spaces[0].models is not None
         assert spaces[0].datasets is not None