Skip to content

Commit 66e9ba1

Browse files
committed
plotting fixes
1 parent c457723 commit 66e9ba1

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

scripts/plotting/create_timing_option_learning_plots.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,13 +105,14 @@ def _load_results_for_seed(df: pd.DataFrame, selector: Callable[[pd.DataFrame],
105105
if np.isnan(result):
106106
continue # timeout
107107
all_time_results.append(result)
108-
assert len(all_time_results) > 0
108+
if not all_time_results:
109+
return [0, TIMEOUT], [0, 0]
109110
seed_xs = [0] + sorted(all_time_results)
110111
seed_ys = list(range(len(seed_xs)))
111112
assert seed_xs[-1] < TIMEOUT
112113
seed_xs.append(TIMEOUT)
113114
seed_ys.append(seed_ys[-1])
114-
seed_ys = np.array(seed_ys) / len(TASK_IDS) # convert to percent solved
115+
seed_ys = 100 * np.array(seed_ys) / len(TASK_IDS) # convert to percent solved
115116
return seed_xs, seed_ys
116117

117118

0 commit comments

Comments
 (0)