Commit 51b866f 1 parent e2bc481 commit 51b866f Copy full SHA for 51b866f
File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -2318,13 +2318,13 @@ def test_list_spaces_with_datasets(self):
2318
2318
assert "wikipedia" in spaces [0 ].datasets
2319
2319
2320
2320
def test_list_spaces_linked (self ):
2321
- space_id = "open-llm-leaderboard/open_llm_leaderboard "
2321
+ space_id = "stabilityai/stable-diffusion "
2322
2322
2323
- spaces = list ( self ._api .list_spaces (search = space_id ))
2323
+ spaces = [ space for space in self ._api .list_spaces (search = space_id ) if space . id == space_id ]
2324
2324
assert spaces [0 ].models is None
2325
2325
assert spaces [0 ].datasets is None
2326
2326
2327
- spaces = list ( self ._api .list_spaces (search = space_id , linked = True ))
2327
+ spaces = [ space for space in self ._api .list_spaces (search = space_id , linked = True ) if space . id == space_id ]
2328
2328
assert spaces [0 ].models is not None
2329
2329
assert spaces [0 ].datasets is not None
2330
2330
You can’t perform that action at this time.
0 commit comments