From e4457400a65e0ea515f9e2e28fff2a681d29b44b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matt=C3=A9o=20Delabre?= <spam@delab.re>
Date: Sun, 15 Dec 2019 23:52:57 -0500
Subject: [PATCH] Add larger set of colors for plot

---
 constellation/util.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/constellation/util.py b/constellation/util.py
index a02c2c0..bc8f649 100644
--- a/constellation/util.py
+++ b/constellation/util.py
@@ -1,5 +1,7 @@
 import torch
 import matplotlib
+from matplotlib.colors import ListedColormap
+import seaborn
 
 
 def get_random_messages(count, order):
@@ -61,7 +63,7 @@ def plot_constellation(
     ax.grid()
 
     order = len(constellation)
-    color_map = matplotlib.cm.Dark2
+    color_map = ListedColormap(seaborn.color_palette('husl', n_colors=order))
     color_norm = matplotlib.colors.BoundaryNorm(range(order + 1), order)
 
     # Extend axes symmetrically around zero so that they fit data