From 3d966b3f7bb58e11e5544365da3ffd9fe13faa7d Mon Sep 17 00:00:00 2001 From: Roy Smart Date: Mon, 11 Nov 2024 15:18:44 -0700 Subject: [PATCH] Relaxed the type constraints in `named_arrays.plt.rgbmesh()`. (#98) --- named_arrays/plt.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/named_arrays/plt.py b/named_arrays/plt.py index f98e01a..0f56e24 100644 --- 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