Add previous changes

dashboard.py

@@ -39,19 +39,8 @@ st.markdown('#')
 with st.spinner(text=f'Checking wandb...'):
     df_runs = io.load_runs(project=DEFAULT_PROJECT, filters=DEFAULT_FILTERS, min_steps=10)
 
-
-### Wandb Runs ###
-# with st.sidebar:
-
-#     st.markdown('#')
-#     st.sidebar.header(":violet[Select] Runs")
-
-    # df_runs_subset = io.filter_dataframe(df_runs, demo_selection=df_runs.id.isin(DEFAULT_SELECTED_RUNS))
-    # n_runs = len(df_runs_subset)
-
 metric.wandb(df_runs)
 
-
 # add vertical space
 st.markdown('#')
 st.markdown('#')
@@ -65,9 +54,6 @@ with tab1:
     st.subheader(":violet[Run] Data")
     with st.expander(f'Show :violet[raw] wandb data'):
 
-        # filter_selected_checkbox = st.checkbox('Filter to selected runs', value=True)
-        # df_to_show = df_runs_subset if filter_selected_checkbox else df_runs
-
         edited_df = st.data_editor(
             df_runs.assign(Select=False).set_index('Select'),
             column_config={"Select": st.column_config.CheckboxColumn(required=True)},

opendashboards/utils/plotting.py

@@ -44,7 +44,7 @@ def plot_throughput(df: pd.DataFrame, n_minutes: int = 10) -> go.Figure:
 
 
 def plot_weights(scores: pd.DataFrame, ntop: int = 20, uids: List[Union[str, int]] = None) -> go.Figure:
-    """_summary_
+    """Plot weights of uids.
 
     Args:
         scores (pd.DataFrame): Dataframe of scores. Should be indexed by timestamp and have one column per uid.
@@ -310,11 +310,15 @@ def plot_completion_length_time(
     uid_col: str = "answer_uids",
     completion_col: str = "answer_completions",
     time_col: str = "answer_times",
+    uids: List[int] = None,
     words: bool = False,
 ) -> go.Figure:
 
     df = df[[uid_col, completion_col, time_col]].explode(column=[uid_col, completion_col, time_col])
     df["time"] = df[time_col].astype(float)
+    if uids is not None:
+        df = df.loc[df[uid_col].isin(uids)]
+        
     if words:
         df["completion_length"] = df[completion_col].str.split().str.len()
     else:
@@ -324,6 +328,7 @@ def plot_completion_length_time(
         df,
         x='completion_length',
         y='time',
+        color=uid_col if uids is not None else None,
         labels={"completion_length": f"Completion Length, {'Words' if words else 'Characters'}", "time": "Time (s)"},
         title=f"Completion Length vs Time, {'Words' if words else 'Characters'}",
         marginal_x="histogram",