Fix tests listing text generation models (#2593)

* Fix tests listing text generation models * mark legacy test as flaky * Revert "Fix tests listing text generation models" This reverts commit 909361c. * Revert "Revert "Fix tests listing text generation motdels"" This reverts commit 62eaab7. * Revert "mark legacy test as flaky" This reverts commit 24e7d2c.
huggingface · Oct 8, 2024 · 39c7a8b · 39c7a8b
1 parent 1b8e9cf
commit 39c7a8b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_inference_async_client.py b/tests/test_inference_async_client.py
@@ -325,7 +325,7 @@ async def test_list_deployed_models_single_frameworks() -> None:
             assert isinstance(model, str)
 
     assert "text-generation" in models_by_task
-    assert "bigscience/bloom" in models_by_task["text-generation"]
+    assert "HuggingFaceH4/zephyr-7b-beta" in models_by_task["text-generation"]
 
 
 @pytest.mark.asyncio

diff --git a/tests/test_inference_client.py b/tests/test_inference_client.py
@@ -859,7 +859,7 @@ def test_list_deployed_models_single_frameworks(self) -> None:
                 self.assertIsInstance(model, str)
 
         self.assertIn("text-generation", models_by_task)
-        self.assertIn("bigscience/bloom", models_by_task["text-generation"])
+        self.assertIn("HuggingFaceH4/zephyr-7b-beta", models_by_task["text-generation"])
 
 
 @pytest.mark.vcr