Fix colormaps registration due to matplotlib > 3.7

gallantlab · May 20, 2024 · b8675a7 · b8675a7
1 parent d8d73c0
commit b8675a7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/cortex/dataset/views.py b/cortex/dataset/views.py
@@ -10,12 +10,13 @@
 
 default_cmap = options.config.get("basic", "default_cmap")
 
+# register_cmap is deprecated in matplotlib > 3.7.0 and replaced by colormaps.register
 try:
-    from matplotlib.cm import register_cmap
-except ImportError:
-    from matplotlib import colormaps
+    from matplotlib import colormaps as cm
     def register_cmap(cmap):
-        return colormaps.register(cmap)
+        return cm.register(cmap)
+except ImportError:
+    from matplotlib.cm import register_cmap
 
 
 def normalize(data):

diff --git a/cortex/utils.py b/cortex/utils.py
@@ -23,12 +23,13 @@
 from .testing_utils import INKSCAPE_VERSION
 from .volume import anat2epispace
 
+# register_cmap is deprecated in matplotlib > 3.7.0 and replaced by colormaps.register
 try:
-    from matplotlib.cm import register_cmap
-except ImportError:
-    from matplotlib import colormaps
+    from matplotlib import colormaps as cm
     def register_cmap(cmap):
-        return colormaps.register(cmap)
+        return cm.register(cmap)
+except ImportError:
+    from matplotlib.cm import register_cmap
 
 
 class DocLoader(object):