Skip to content

Commit e9319e1

Browse files
committed
hotfix autotrain
1 parent 8d7005e commit e9319e1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

code/util/plot_autotrain_manager.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,8 @@ def plot_manager_all_progress_bokeh_source(
132132

133133
# Append subjects that have not graduated
134134
subject_ids = subject_ids + [s for s in df_manager.subject_id.unique() if s not in subject_ids]
135+
# Only subjects in df_to_draw
136+
subject_ids = [s for s in subject_ids if s in df_to_draw.subject_id.unique()]
135137
else:
136138
raise ValueError("Invalid sort_by value.")
137139

@@ -241,7 +243,7 @@ def plot_manager_all_progress_bokeh(
241243
),
242244
x_axis_label=x_axis,
243245
y_axis_label="Subjects",
244-
height=770 + 20 * len(subject_ids),
246+
height=max(770, 30 * len(subject_ids)),
245247
width=1400,
246248
# tools=[hover, "lasso_select", "reset", "tap", "pan", "wheel_zoom"],
247249
# tooltips=TOOLTIPS,

0 commit comments

Comments
 (0)