From 0e08e1d8b4cbf7cc8af8105140a9f6a5ecbe7183 Mon Sep 17 00:00:00 2001 From: Lucain Date: Tue, 1 Oct 2024 14:49:42 +0200 Subject: [PATCH] Fix test (#2582) --- tests/test_hf_api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py index a028066847..4328e05d76 100644 --- 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):