From 30d9c3fdc3637b2f121ae03cf9cd5758c665f005 Mon Sep 17 00:00:00 2001 From: Harutaka Kawamura Date: Wed, 22 May 2024 16:37:53 +0900 Subject: [PATCH] Change `MLFLOW_ENABLE_TRACE_IN_SERVING` default value to False (#12095) Signed-off-by: harupy <17039389+harupy@users.noreply.github.com> --- mlflow/environment_variables.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/mlflow/environment_variables.py b/mlflow/environment_variables.py index 1c9878c8087df..956f37bb363a6 100644 --- a/mlflow/environment_variables.py +++ b/mlflow/environment_variables.py @@ -579,9 +579,12 @@ def get(self): # How many traces to be buffered at the in-memory trace client. MLFLOW_TRACE_BUFFER_MAX_SIZE = _EnvironmentVariable("MLFLOW_TRACE_BUFFER_MAX_SIZE", int, 1000) -# Whether or not to enable trace logging in MLflow in served models. -# TODO: Document the behavior of this flag. -MLFLOW_ENABLE_TRACE_IN_SERVING = _BooleanEnvironmentVariable("MLFLOW_ENABLE_TRACE_IN_SERVING", True) +#: Whether or not to enable trace logging in model serving. +#: The default value is set to False to ensure that this flag is only enabled +#: when our internal safety mechanism on Databricks explicitly sets it to True. +MLFLOW_ENABLE_TRACE_IN_SERVING = _BooleanEnvironmentVariable( + "MLFLOW_ENABLE_TRACE_IN_SERVING", False +) # Whether to use presigned URLs to interact with the Unity Catalog MLFLOW_UNITY_CATALOG_PRESIGNED_URLS_ENABLED = _BooleanEnvironmentVariable(