check for matplotlibs.colormaps before matplotlibs.colormaps.register

JuliaPy · Jul 5, 2024 · bd9e8a8 · bd9e8a8
1 parent aff0598
commit bd9e8a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/colormaps.jl b/src/colormaps.jl
@@ -49,7 +49,7 @@ function init_colormaps()
     PythonCall.pycopy!(LinearSegmentedColormap, colorsm.LinearSegmentedColormap)
 
     PythonCall.pycopy!(cm_get_cmap, pyhasattr(pyplot, "get_cmap") ? pyplot.get_cmap : cm.get_cmap)
-    PythonCall.pycopy!(cm_register_cmap, pyhasattr(matplotlib.colormaps, "register") ? matplotlib.colormaps.register : cm.register_cmap)
+    PythonCall.pycopy!(cm_register_cmap, pyhasattr(matplotlib, "colormaps") && pyhasattr(matplotlib.colormaps, "register") ? matplotlib.colormaps.register : cm.register_cmap)
 
     PythonCall.pycopy!(ScalarMappable, cm.ScalarMappable)
     PythonCall.pycopy!(Normalize01, pycall(colorsm.Normalize; vmin=0,vmax=1))