From f9766959a25367d659973a57e698d73e43183fa5 Mon Sep 17 00:00:00 2001 From: Lucain Pouget Date: Tue, 1 Oct 2024 14:41:11 +0200 Subject: [PATCH] Fix test --- 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):