Update plotting.py

agosztolai · Jun 28, 2023 · e354333 · e354333
1 parent fcc9292
commit e354333
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MARBLE/plotting.py b/MARBLE/plotting.py
@@ -237,7 +237,7 @@ def embedding(
                         ax.scatter(emb_[t, 0], emb_[t, 1], emb_[t, 2], c=cgrad, alpha=alpha, s=s, label=title)  
         else:
             if dim == 2:
-                ax.scatter(emb_[:, 0], emb_[:, 1], c=c, alpha=alpha, s=s, label=title)
+                ax.scatter(emb_[:, 0], emb_[:, 1], c=c_, alpha=alpha, s=s, label=title)
             elif dim == 3:
                 ax.scatter(emb_[:, 0], emb_[:, 1], emb_[:, 2], c=c, alpha=alpha, s=s, label=title)