moved cmap based on version

VlachosGroup · May 16, 2024 · 2f8b3d3 · 2f8b3d3
1 parent 895dfcb
commit 2f8b3d3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/aimsim/tasks/cluster_data.py b/aimsim/tasks/cluster_data.py
@@ -3,7 +3,10 @@
 from os.path import dirname
 
 import matplotlib.pyplot as plt
-import matplotlib.cm as colormaps
+try:
+    import matplotlib.cm import get_cmap
+except:
+    from matplotlib.pyplot import get_cmap
 from matplotlib.colors import rgb2hex
 import yaml