use newer mpl.colormaps object

DHI · Jun 4, 2024 · b8a1b53 · b8a1b53
1 parent 4290edf
commit b8a1b53
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/terracotta/cmaps/generate_cmaps.py b/terracotta/cmaps/generate_cmaps.py
@@ -4,11 +4,11 @@
 """
 
 import numpy as np
-import matplotlib.cm as cm
+import matploblib as mpl
 
 from terracotta.cmaps.get_cmaps import SUFFIX
 
-ALL_CMAPS = list(cm.cmaps_listed) + list(cm.datad)
+ALL_CMAPS = list(mpl.colormaps)
 ALL_CMAPS.extend([f"{cmap}_r" for cmap in ALL_CMAPS])
 NUM_VALS = 255
 
@@ -17,7 +17,7 @@ def generate_maps(out_folder: str) -> None:
     x = np.linspace(0, 1, NUM_VALS)
     for cmap in ALL_CMAPS:
         print(cmap)
-        cmap_fun = cm.get_cmap(cmap)
+        cmap_fun = mpl.colormaps[cmap]
         cmap_vals = cmap_fun(x)
         cmap_uint8 = (cmap_vals * 255).astype("uint8")
         np.save(f"{out_folder}/{cmap.lower()}{SUFFIX}", cmap_uint8)