Fix test (#2582)

huggingface · Oct 1, 2024 · 0e08e1d · 0e08e1d
1 parent 476fa0b
commit 0e08e1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py
@@ -1754,11 +1754,11 @@ def test_list_models_complex_query(self):
         assert all(tag in model.tags for tag in ["bert", "jax"])
 
     def test_list_models_with_config(self):
-        for model in self._api.list_models(filter="adapter-transformers", fetch_config=True, limit=20):
+        for model in self._api.list_models(filter=("adapter-transformers", "bert"), fetch_config=True, limit=20):
             self.assertIsNotNone(model.config)
 
     def test_list_models_without_config(self):
-        for model in self._api.list_models(filter="adapter-transformers", fetch_config=False, limit=20):
+        for model in self._api.list_models(filter=("adapter-transformers", "bert"), fetch_config=False, limit=20):
             self.assertIsNone(model.config)
 
     def test_list_models_expand_author(self):