Merge branch 'main' into deprecate-hf-inference-specific-features

huggingface · Feb 12, 2025 · 411486e · 411486e
2 parents 2337a56 + 85694fe
commit 411486e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py
@@ -1917,10 +1917,12 @@ def test_list_models_non_gated_only(self):
         for model in self._api.list_models(expand=["gated"], gated=False, limit=5):
             assert model.gated is False
 
+    @pytest.mark.skip("Inference parameter is being revamped")
     def test_list_models_inference_warm(self):
         for model in self._api.list_models(inference=["warm"], expand="inference", limit=5):
             assert model.inference == "warm"
 
+    @pytest.mark.skip("Inference parameter is being revamped")
     def test_list_models_inference_cold(self):
         for model in self._api.list_models(inference=["cold"], expand="inference", limit=5):
             assert model.inference == "cold"