diff --git a/src/anomalib/loggers/mlflow.py b/src/anomalib/loggers/mlflow.py index 64504893d4..f98723e6f4 100644 --- a/src/anomalib/loggers/mlflow.py +++ b/src/anomalib/loggers/mlflow.py @@ -1,5 +1,6 @@ """MLFlow logger with add image interface.""" +import os from typing import Literal import numpy as np @@ -65,7 +66,7 @@ def __init__( self, experiment_name: str | None = "anomalib_logs", run_name: str | None = None, - tracking_uri: str | None = None, + tracking_uri: str | None = os.getenv("MLFLOW_TRACKING_URI"), save_dir: str | None = "./mlruns", log_model: Literal[True, False, "all"] | None = False, prefix: str | None = "",