diff --git a/src/cmap/_color.py b/src/cmap/_color.py index db57f085d..56cde019b 100644 --- a/src/cmap/_color.py +++ b/src/cmap/_color.py @@ -46,6 +46,7 @@ RGBTuple, # 3-tuple of all ints or all floats RGBATuple, # 4-tuple of all floats, or 3 ints and 1 float np.ndarray, # 3- or 4-element rgb(a) vector + np.number, list[Union[float, int]], # 3- or 4-element rgb(a) vector "Color", # another color object ] diff --git a/src/cmap/_colormap.py b/src/cmap/_colormap.py index 18a9ff5f7..ba4712891 100644 --- a/src/cmap/_colormap.py +++ b/src/cmap/_colormap.py @@ -1077,7 +1077,7 @@ def to_css( If `True`, return colors as hex strings, by default use `rgba()` strings. """ if max_stops and len(self._stops) > max_stops: - stops = tuple(np.linspace(0, 1, max_stops)) + stops: tuple[np.floating, ...] = tuple(np.linspace(0, 1, max_stops)) colors = tuple(Color(c) for c in self.to_lut(max_stops)) else: stops, colors = self.stops, self.colors