Support old matplotlib by matplotlib.cm.get_cmap

wkentaro · Feb 14, 2023 · d8b8357 · d8b8357
1 parent c55bfb6
commit d8b8357
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/imgviz/depth.py b/imgviz/depth.py
@@ -68,7 +68,10 @@ def __call__(self, depth, dtype=np.uint8):
         normalized[isnan] = 0
 
         if isinstance(self._colormap, str):
-            colormap_func = matplotlib.colormaps[self._colormap]
+            if hasattr(matplotlib, "colormaps"):
+                colormap_func = matplotlib.colormaps[self._colormap]
+            else:
+                colormap_func = matplotlib.cm.get_cmap(self._colormap)
         else:
             colormap_func = self._colormap
         rgb = colormap_func(normalized)[:, :, :3]