Skip to content

Commit 0e08e1d

Browse files
authored
Fix test (#2582)
1 parent 476fa0b commit 0e08e1d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_hf_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1754,11 +1754,11 @@ def test_list_models_complex_query(self):
17541754
assert all(tag in model.tags for tag in ["bert", "jax"])
17551755

17561756
def test_list_models_with_config(self):
1757-
for model in self._api.list_models(filter="adapter-transformers", fetch_config=True, limit=20):
1757+
for model in self._api.list_models(filter=("adapter-transformers", "bert"), fetch_config=True, limit=20):
17581758
self.assertIsNotNone(model.config)
17591759

17601760
def test_list_models_without_config(self):
1761-
for model in self._api.list_models(filter="adapter-transformers", fetch_config=False, limit=20):
1761+
for model in self._api.list_models(filter=("adapter-transformers", "bert"), fetch_config=False, limit=20):
17621762
self.assertIsNone(model.config)
17631763

17641764
def test_list_models_expand_author(self):

0 commit comments

Comments
 (0)