From dcf3ef440a2c81d5a0805c7b0efd0edaca8e9eca Mon Sep 17 00:00:00 2001
From: Victor de Souza magalhaes <vdesouza@nrao.edu>
Date: Wed, 4 Jun 2025 09:48:23 -0600
Subject: [PATCH] simple_imshow_map_plot now uses get_proper_color_map for
 colormap resolution.

---
 src/astrohack/visualization/plot_tools.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/astrohack/visualization/plot_tools.py b/src/astrohack/visualization/plot_tools.py
index 665501c5..2c93b266 100644
--- a/src/astrohack/visualization/plot_tools.py
+++ b/src/astrohack/visualization/plot_tools.py
@@ -380,7 +380,7 @@ def simple_imshow_map_plot(
     y_axis,
     gridded_2d_arr,
     title,
-    cmap,
+    colormap,
     zlim,
     x_label="X axis [m]",
     y_label="Y axis [m]",
@@ -390,6 +390,7 @@ def simple_imshow_map_plot(
     extent_margin=0,
     add_colorbar=True,
 ):
+    cmap = get_proper_color_map(colormap)
     if zlim is None:
         minmax = [np.nanmin(gridded_2d_arr), np.nanmax(gridded_2d_arr)]
     else: