Relaxed the type constraints in named_arrays.plt.rgbmesh(). (#98)

sun-data · Nov 11, 2024 · 3d966b3 · 3d966b3
1 parent 9dc13dd
commit 3d966b3
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/named_arrays/plt.py b/named_arrays/plt.py
@@ -880,14 +880,20 @@ def rgbmesh(
 
     if len(WXY) == 1:
         WXY, = WXY
-        if isinstance(WXY, na.AbstractSpectralPositionalVectorArray):
+        if isinstance(WXY, na.AbstractSpectralVectorArray):
             w = WXY.wavelength
-            x = WXY.position.x
-            y = WXY.position.y
+            if isinstance(WXY, na.AbstractPositionalVectorArray):
+                x = WXY.position.x
+                y = WXY.position.y
+            else:  # pragma: nocover
+                raise TypeError(
+                    "if one positional argument, it must be an instance of "
+                    f"`na.AbstractPositionalVectorArray`, got {type(WXY)}."
+                )
         else:   # pragma: nocover
             raise TypeError(
                 "if one positional argument, it must be an instance of "
-                f"`na.AbstractSpectralPositionalVectorArray`, got {type(WXY)}."
+                f"`na.AbstractSpectralVectorArray`, got {type(WXY)}."
             )
     elif len(WXY) == 2:
         w, XY = WXY