diff --git a/tests/hermes_cli/test_model_validation.py b/tests/hermes_cli/test_model_validation.py index 9e08831701e3..cfff3080ebfa 100644 --- a/tests/hermes_cli/test_model_validation.py +++ b/tests/hermes_cli/test_model_validation.py @@ -949,7 +949,7 @@ class TestProbeApiModelsUserAgent: body = b'{"data":[]}' with patch( - "hermes_cli.models.urllib.request.urlopen", + "hermes_cli.models._urlopen_model_catalog_request", return_value=self._make_mock_response(body), ) as mock_urlopen: probe_api_models( @@ -965,7 +965,7 @@ class TestProbeApiModelsUserAgent: body = b'{"data":[]}' with patch( - "hermes_cli.models.urllib.request.urlopen", + "hermes_cli.models._urlopen_model_catalog_request", return_value=self._make_mock_response(body), ) as mock_urlopen: probe_api_models("provider-key", "https://api.example.com/v1")