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