901cc588fee90215d2e2831bd4a3e7481b307ba7,slm_lab/experiment/analysis.py,,plot_session,#Any#Any#Any#,106

Before Change

    if aeb_count <= 8:
        palette = cl.scales[str(max(3, aeb_count))]["qual"]["Set2"]
    else:
        palette = util.interp(cl.scales["8"]["qual"]["Set2"], aeb_count)
    fig = viz.tools.make_subplots(rows=3, cols=1, shared_xaxes=True)
    for idx, (a, e, b) in enumerate(session_data):
        aeb_str = f"{a}{e}{b}"

After Change

    for idx, (a, e, b) in enumerate(session_data):
        aeb_str = f"{a}{e}{b}"
        aeb_df = session_data[(a, e, b)]
        aeb_df.fillna(0, inplace=True)  // for saving plot, cant have nan
        fig_1 = viz.plot_line(aeb_df, "reward", "epi", legend_name=aeb_str, draw=False, trace_kwargs={"legendgroup": aeb_str, "line": {"color": palette[idx]}})
        fig.append_trace(fig_1.data[0], 1, 1)

Instances