From 8ca842968035ff1856addf07fa145859a1b0e383 Mon Sep 17 00:00:00 2001 From: Harutaka Kawamura Date: Wed, 22 Nov 2023 20:54:33 +0900 Subject: [PATCH] Set timeout default to `None` (#10487) Signed-off-by: harupy <17039389+harupy@users.noreply.github.com> --- mlflow/deployments/databricks/__init__.py | 4 ++-- mlflow/deployments/mlflow/__init__.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mlflow/deployments/databricks/__init__.py b/mlflow/deployments/databricks/__init__.py index 67c5f85c242df..60b79b9281c13 100644 --- a/mlflow/deployments/databricks/__init__.py +++ b/mlflow/deployments/databricks/__init__.py @@ -70,7 +70,7 @@ def _call_endpoint( prefix: str = "/api/2.0", route: Optional[str] = None, json_body: Optional[Dict[str, Any]] = None, - timeout: int = MLFLOW_HTTP_REQUEST_TIMEOUT.get(), + timeout: Optional[int] = None, ): call_kwargs = {} if method.lower() == "get": @@ -82,7 +82,7 @@ def _call_endpoint( host_creds=get_databricks_host_creds(self.target_uri), endpoint=posixpath.join(prefix, "serving-endpoints", route or ""), method=method, - timeout=timeout, + timeout=MLFLOW_HTTP_REQUEST_TIMEOUT.get() if timeout is None else timeout, raise_on_status=False, retry_codes=MLFLOW_DEPLOYMENT_CLIENT_REQUEST_RETRY_CODES, **call_kwargs, diff --git a/mlflow/deployments/mlflow/__init__.py b/mlflow/deployments/mlflow/__init__.py index 14bc5653fddbe..021995a1e9bf6 100644 --- a/mlflow/deployments/mlflow/__init__.py +++ b/mlflow/deployments/mlflow/__init__.py @@ -93,7 +93,7 @@ def _call_endpoint( method: str, route: str, json_body: Optional[str] = None, - timeout: int = MLFLOW_HTTP_REQUEST_TIMEOUT.get(), + timeout: Optional[int] = None, ): call_kwargs = {} if method.lower() == "get": @@ -105,7 +105,7 @@ def _call_endpoint( host_creds=_get_default_host_creds(self.target_uri), endpoint=route, method=method, - timeout=timeout, + timeout=MLFLOW_HTTP_REQUEST_TIMEOUT.get() if timeout is None else timeout, retry_codes=MLFLOW_DEPLOYMENT_CLIENT_REQUEST_RETRY_CODES, raise_on_status=False, **call_kwargs,