Updated named_arrays.plt.scatter() to handle matplotlib >= 3.8

sun-data · Jan 15, 2024 · 46770a6 · 46770a6
1 parent 3a07bec
commit 46770a6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/named_arrays/_scalars/scalar_named_array_functions.py b/named_arrays/_scalars/scalar_named_array_functions.py
@@ -385,7 +385,10 @@ def plt_scatter(
         s = na.broadcast_to(s, shape_orthogonal)
 
     if np.issubdtype(na.get_dtype(c), np.number):
-        c = na.broadcast_to(c, shape)
+        if "rgba" in c.shape:
+            c = na.broadcast_to(c, shape | dict(rgba=c.shape["rgba"]))
+        else:
+            c = na.broadcast_to(c, shape)
     else:
         if "rgba" in c.shape:
             c = na.broadcast_to(c, shape_orthogonal | dict(rgba=c.shape["rgba"]))