Skip to content

Commit d094d4b

Browse files
CarlosNacherCarlos Náchersamet-akcay
authored
fix: get MLFLOW_TRACKING_UTI from env variables as default (#2107)
* fix: get MLFLOW_TRACKING_UTI from env variables as default In `AnomalibMLFlowLogger`, `tracking_uri` was set to None, which was overriding the parameter in the underlying pytorch `MLFloeLogger` so never could log into remote server. This fixes it. * Update src/anomalib/loggers/mlflow.py * Update mlflow.py --------- Co-authored-by: Carlos Nácher <cnacher@nutai.com> Co-authored-by: Samet Akcay <samet.akcay@intel.com>
1 parent e7e6cb8 commit d094d4b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/anomalib/loggers/mlflow.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
"""MLFlow logger with add image interface."""
22

3+
import os
34
from typing import Literal
45

56
import numpy as np
@@ -65,7 +66,7 @@ def __init__(
6566
self,
6667
experiment_name: str | None = "anomalib_logs",
6768
run_name: str | None = None,
68-
tracking_uri: str | None = None,
69+
tracking_uri: str | None = os.getenv("MLFLOW_TRACKING_URI"),
6970
save_dir: str | None = "./mlruns",
7071
log_model: Literal[True, False, "all"] | None = False,
7172
prefix: str | None = "",

0 commit comments

Comments
 (0)