Fix #36

Marsilea-viz · Jul 3, 2024 · 229d7c5 · 229d7c5
1 parent 5b75622
commit 229d7c5
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/marsilea/utils.py b/src/marsilea/utils.py
@@ -69,15 +69,13 @@ def relative_luminance(color):
 
 
 def get_colormap(cmap):
-    if isinstance(cmap, Colormap):
-        return cmap
     try:
-        return mpl.colormap.get(cmap)
+        return mpl.cm.ColormapRegistry.get_cmap(cmap)
     except AttributeError:
         try:
-            return mpl.cm.get_cmap(cmap)
+            return mpl.colormaps.get(cmap)
         except AttributeError:
-            return mpl.cm.ColormapRegistry.get_cmap(cmap)
+            return mpl.cm.get_cmap(cmap)
 
 
 def get_canvas_size_by_data(