revert decorators order

huggingface · Feb 13, 2025 · bd6d856 · bd6d856
1 parent 25ef0db
commit bd6d856
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/test_inference_client.py b/tests/test_inference_client.py
@@ -43,6 +43,7 @@
     TranslationOutput,
     VisualQuestionAnsweringOutputElement,
     ZeroShotClassificationOutputElement,
+    constants,
     hf_hub_download,
 )
 from huggingface_hub.errors import HfHubHTTPError, ValidationError
@@ -209,7 +210,7 @@ def list_clients(task: str) -> List[pytest.param]:
     return clients
 
 
-@pytest.fixture
+@pytest.fixture(scope="module")
 @with_production_testing
 def client(request):
     """
@@ -266,8 +267,8 @@ def mock_fetch():
         monkeypatch.setattr("huggingface_hub.inference._providers.hf_inference._fetch_recommended_models", mock_fetch)
 
 
+@pytest.mark.vcr
 @with_production_testing
-@pytest.mark.vcr()
 class TestInferenceClient(TestBase):
     @pytest.mark.parametrize("client", list_clients("audio-classification"), indirect=True)
     def test_audio_classification(self, client: InferenceClient):
@@ -903,16 +904,12 @@ class TestListDeployedModels(TestBase):
     @expect_deprecation("list_deployed_models")
     @patch("huggingface_hub.inference._client.get_session")
     def test_list_deployed_models_main_frameworks_mock(self, get_session_mock: MagicMock) -> None:
-        from huggingface_hub import constants
-
         InferenceClient().list_deployed_models()
         assert len(get_session_mock.return_value.get.call_args_list) == len(constants.MAIN_INFERENCE_API_FRAMEWORKS)
 
     @expect_deprecation("list_deployed_models")
     @patch("huggingface_hub.inference._client.get_session")
     def test_list_deployed_models_all_frameworks_mock(self, get_session_mock: MagicMock) -> None:
-        from huggingface_hub import constants
-
         InferenceClient().list_deployed_models("all")
         assert len(get_session_mock.return_value.get.call_args_list) == len(constants.ALL_INFERENCE_API_FRAMEWORKS)