Change colors

utils.py

@@ -30,8 +30,8 @@ def dataframe_to_pianoroll_img(df):
     fig = plt.figure(figsize=(8, 5))
     ax = fig.add_subplot(111)
     ax.scatter(df.start_time, df.pitch, c="white")
-    print(mcolors.TABLEAU_COLORS)
-    colordict = {inst: mcolors.TABLEAU_COLORS[i] for i, inst in enumerate(df["instrument"].unique())}
+    colors = [mcolors.TABLEAU_COLORS[name] for i, name in enumerate(list(mcolors.TABLEAU_COLORS))]
+    colordict = {inst: colors[i] for i, inst in enumerate(df["instrument"].unique())}
     for _, row in df.iterrows():
         ax.add_patch(Rectangle((row["start_time"],
                                 row["pitch"]-0.4),