update deprecated cmap retrival

dpeerlab · Nov 16, 2024 · 41d5fa8 · 41d5fa8
1 parent 81fa893
commit 41d5fa8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/palantir/plot.py b/src/palantir/plot.py
@@ -1155,8 +1155,11 @@ def plot_stats(
     }
     scatter_kwargs.update(kwargs)
 
-    default_cmap = matplotlib.colormaps["viridis"]
-    cmap = copy(matplotlib.colormaps.get(cmap, default_cmap))
+    try:
+        cmap = matplotlib.colormaps[cmap] if isinstance(cmap, str) else cmap
+    except KeyError:
+        cmap = matplotlib.colormaps["viridis"]
+    cmap = copy(cmap)
     cmap.set_bad(na_color)
 
     na_color = matplotlib.colors.to_hex(na_color, keep_alpha=True)