Merge pull request #41 from BlueBrain/update-matplotlib

remove deprecated get_cmap
BlueBrain · May 29, 2024 · 6d288a6 · 6d288a6
2 parents 29753e8 + 2d5fe03
commit 6d288a6
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/currentscape/plotting.py b/currentscape/plotting.py
@@ -483,9 +483,9 @@ def get_colors(cmap, n_col):
         return x.mpl_colors[:n_col], False
 
     # is from matplotlib
-    new_cmap = matplotlib.cm.get_cmap(cmap, n_col)
+    new_cmap = matplotlib.colormaps[cmap].resampled(n_col)
     # check number of colors
-    if matplotlib.cm.get_cmap(cmap).N < n_col:
+    if matplotlib.colormaps[cmap].N < n_col:
         return list(new_cmap(range(n_col))), True
 
     return list(new_cmap(range(n_col))), False

diff --git a/tests/test_currentscape_from_paper.py b/tests/test_currentscape_from_paper.py
@@ -1,4 +1,5 @@
 """Test that example produces a figure."""
+
 import os
 import sys
 from tests.utils import cwd

diff --git a/tests/test_use_case_example.py b/tests/test_use_case_example.py
@@ -1,4 +1,5 @@
 """Test that example produces a figure."""
+
 import os
 import subprocess
 import sys

diff --git a/tests/utils.py b/tests/utils.py
@@ -1,4 +1,5 @@
 """Utils."""
+
 import os
 from contextlib import contextmanager