check for matplotlibs.colormaps before matplotlibs.colormaps.register (…

…#592)
JuliaPy · Jul 5, 2024 · c229e30 · c229e30
1 parent 28826e0
commit c229e30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/colormaps.jl b/src/colormaps.jl
@@ -55,7 +55,7 @@ function init_colormaps()
     copy!(LinearSegmentedColormap, colorsm."LinearSegmentedColormap")
 
     copy!(cm_get_cmap, haskey(plt, "get_cmap") ? plt."get_cmap" : cm."get_cmap")
-    copy!(cm_register_cmap, haskey(matplotlib.colormaps, "register") ? matplotlib.colormaps."register" : cm."register_cmap")
+    copy!(cm_register_cmap, haskey(matplotlib, "colormaps") && haskey(matplotlib."colormaps", "register") ? matplotlib."colormaps"."register" : cm."register_cmap")
 
     copy!(ScalarMappable, cm."ScalarMappable")
     copy!(Normalize01, pycall(colorsm."Normalize",PyAny,vmin=0,vmax=1))