Fix for python 3.9

jahn · Apr 11, 2024 · b67bd52 · b67bd52
1 parent f9185cc
commit b67bd52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/python/MITgcmutils/MITgcmutils/cs/pcol.py b/utils/python/MITgcmutils/MITgcmutils/cs/pcol.py
@@ -180,7 +180,7 @@ def pcol( x, y, data, projection=None, vmin=None, vmax=None, **kwargs):
         # add a reasonable colormap
         m = cm.ScalarMappable(cmap=colmap)
         m.set_array(data)
-        plt.colorbar(m)
+        plt.colorbar(m, ax=ax)
     elif mapit == 0:
         ax = fig.axes[-1]
         ax.axis('image')