Don't use deprecated cm.get_cmap in new matplotlib

salilab · Feb 29, 2024 · bfa36bd · bfa36bd
1 parent f423416
commit bfa36bd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/modules/spatiotemporal/pyext/src/write_output.py b/modules/spatiotemporal/pyext/src/write_output.py
@@ -8,7 +8,10 @@
 except ImportError:
     Digraph = None
 try:
-    from matplotlib import cm
+    try:
+        from matplotlib import colormaps as cm  # matplotlib 3.7+
+    except ImportError:
+        from matplotlib import cm
     from matplotlib import colors as clr
 except ImportError:
     cm = None