15 minfix

app.py

@@ -2,9 +2,9 @@ import streamlit as st
 import wandb
 import pandas as pd
 import os
-import matplotlib.pyplot as plt
+# import matplotlib.pyplot as plt
 
-from utils import fetch_runs_to_df
+from utils import fetch_runs_to_df, fetch_run
 
 # Access the API key from the environment variable
 wandb_api_key = os.getenv('WANDB_API_KEY')
@@ -63,9 +63,13 @@ if tag_filter:
 st.dataframe(df)
 
 # Display details of selected run
-selected_run_id = st.selectbox("Select a Run ID to see details", df["ID"].tolist() if not df.empty else [])
-if selected_run_id:
-    selected_run = api.run(f"{projects[selected_project]['entity']}/{projects[selected_project]['project']}/{selected_run_id}")
-    run_df = selected_run.history()
-    st.write(f"Details for run: {selected_run.name}")
-    st.dataframe(run_df)
+if not df.empty:
+    selected_run_id = st.selectbox("Select a Run ID to see details", df["ID"].tolist())
+    
+    run = fetch_run(api, projects, selected_project, selected_run_id)
+    if run:
+        run_df = run.history()
+        st.write(f"Details for run: {run.name}")
+        st.dataframe(run_df)
+else:
+    st.warning("No runs available to select.")

utils.py

@@ -1,9 +1,10 @@
 import streamlit as st
 import pandas as pd
+import wandb
 
 def fetch_runs_to_df(api, projects, selected_project):
     data = []
-    
+
     if selected_project == "All":
         # return all runs from all projects
         for project_name, details in projects.items():
@@ -36,3 +37,24 @@ def fetch_runs_to_df(api, projects, selected_project):
         df = pd.DataFrame(data)
 
     return df
+
+def fetch_run(api, projects, selected_project, selected_run_id):
+    # Fetch run details based on the selected project
+    if selected_project == "All":
+        # Find the project for the selected run_id
+        for project_name, details in projects.items():
+            entity = details["entity"]
+            project = details["project"]
+            try:
+                run = api.run(f"{entity}/{project}/{selected_run_id}")
+                break
+            except wandb.errors.CommError:
+                continue
+        else:
+            st.error(f"Run ID {selected_run_id} not found in any project.")
+    else:
+        entity = projects[selected_project]["entity"]
+        project = projects[selected_project]["project"]
+        run = api.run(f"{entity}/{project}/{selected_run_id}")
+
+    return run