Skip to content

Commit feb6e66

Browse files
committed
fix list_space_author test
1 parent 8831137 commit feb6e66

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

tests/test_hf_api.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1985,13 +1985,10 @@ def test_list_spaces_full(self):
19851985
self.assertTrue(any(space.card_data for space in spaces))
19861986

19871987
def test_list_spaces_author(self):
1988-
spaces = list(self._api.list_spaces(author="evaluate-metric"))
1989-
self.assertGreater(len(spaces), 10)
1990-
self.assertTrue(
1991-
set([space.id for space in spaces]).issuperset(
1992-
set(["evaluate-metric/trec_eval", "evaluate-metric/perplexity"])
1993-
)
1994-
)
1988+
spaces = list(self._api.list_spaces(author="julien-c"))
1989+
assert len(spaces) > 10
1990+
for space in spaces:
1991+
assert space.id.startswith("julien-c/")
19951992

19961993
def test_list_spaces_search(self):
19971994
spaces = list(self._api.list_spaces(search="wikipedia"))

0 commit comments

Comments
 (0)