From 31629f8f6c358bab1fbcec734a8927cef50f33f0 Mon Sep 17 00:00:00 2001 From: Lucain Pouget Date: Wed, 11 Dec 2024 13:44:39 +0100 Subject: [PATCH] Fix test_list_spaces_linked --- tests/test_hf_api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py index d7a583437b..fe2f42681f 100644 --- 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