experiment name

ergodicio · Nov 5, 2024 · 92c5cbc · 92c5cbc
1 parent ab1a9e6
commit 92c5cbc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/deploy.yaml b/.github/workflows/deploy.yaml
@@ -8,7 +8,6 @@ on:
   push:
     branches:
       - main
-      - bugfix/docker
 
   # Allows you to run this workflow manually from the Actions tab
   workflow_dispatch:

diff --git a/tsadar_app.py b/tsadar_app.py
@@ -3,7 +3,7 @@
 config.update("jax_enable_x64", True)
 
 import streamlit as st
-import yaml, os
+import yaml, os, mlflow
 from flatten_dict import flatten, unflatten
 
 from tsadar_gui import config
@@ -81,6 +81,7 @@ def process_file(file_path):
         with c2:
             if st.button("Fit"):
                 # run and wait for the results
+                mlflow.set_experiment(cfg["mlflow"]["experiment"])
                 run_id = run_for_app(cfg, mode="fit")
 
                 st.write(f"The results can be found at the mlflow run id {run_id}")