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.

Author: Wauplin

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

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