fix: tentativa de correcao de variavel

heltonricardo · Dec 27, 2023 · 640d58b · 640d58b
1 parent fa2b786
commit 640d58b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/train.py b/src/train.py
@@ -126,6 +126,8 @@ def config_mlflow():
         None
     """
 
+    os.environ['MLFLOW_TRACKING_USERNAME'] = os.getenv("MLFLOW_TRACKING_USERNAME")
+    os.environ['MLFLOW_TRACKING_PASSWORD'] = os.getenv("MLFLOW_TRACKING_PASSWORD")
     mlflow.set_tracking_uri(os.getenv("MLFLOW_TRACKING_URI"))
     mlflow.tensorflow.autolog(
         log_models=True, log_input_examples=True, log_model_signatures=True