d490edce8fc483707c58e852286950ace7e8cb59,batchflow/utils.py,,show_research,#Any#Any#Any#Any#Any#Any#Any#,132

Before Change

        color = list(mcolors.TABLEAU_COLORS.keys())
    df_len = len(df["config"].unique())
    replace = not len(color) > df_len
    chosen_colors = np.random.choice(color, replace=replace, size=df_len)

    kwargs = {"figsize": (9 * len(layouts), 7), "nrows": 1, "ncols": len(layouts), **kwargs}

    _, ax = plt.subplots(**kwargs)

After Change

    if isinstance(color, str):
        cmap = plt.get_cmap(color)
        chosen_colors = [cmap(i/df_len) for i in range(df_len)]
    else:
        chosen_colors = itertools.cycle(color)

    kwargs = {"figsize": (9 * len(layouts), 7), "nrows": 1, "ncols": len(layouts), **kwargs}

Instances