diff --git a/mlflow_adsp/services/endpoint_manager.py b/mlflow_adsp/services/endpoint_manager.py index ed615eb..66fac1a 100644 --- a/mlflow_adsp/services/endpoint_manager.py +++ b/mlflow_adsp/services/endpoint_manager.py @@ -103,7 +103,7 @@ def _poll_network_service(self, process: subprocess.Popen) -> bool: try: version_endpoint_url: str = f"http://{self.params.host}:{self.params.port}/version" - response: Response = requests.get(url=version_endpoint_url, timeout=1) + response: Response = requests.get(url=version_endpoint_url) if response.status_code != 200: logger.debug("Service not yet healthy, waiting ..") EndpointManager._proc_comm(process=process, timeout=self.params.timeout) diff --git a/test/unit/services/test_endpoint_manager.py b/test/unit/services/test_endpoint_manager.py index dcf23bf..b6ed182 100644 --- a/test/unit/services/test_endpoint_manager.py +++ b/test/unit/services/test_endpoint_manager.py @@ -386,7 +386,7 @@ def test_poll_network_service_gracefully_fails(monkeypatch): client: MagicMock = MagicMock() monkeypatch.setattr(subprocess, "Popen", MockPOpen) - def mock_get(url, timeout): + def mock_get(url): raise requests.exceptions.ConnectionError("Boom!") monkeypatch.setattr(requests, "get", mock_get)