Fixed a bug in colorsynth.colorbar() where the wavelength axis wasn…

…'t being selected correctly.
sun-data · Nov 5, 2024 · defc3f4 · defc3f4
1 parent bd723a9
commit defc3f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/colorsynth/_colorsynth.py b/colorsynth/_colorsynth.py
@@ -979,8 +979,8 @@ def colorbar(
 
     spd_, wavelength_ = transform_spd_wavelength(cbar, wavelength)
 
-    XYZ = XYZcie1931_from_spd(spd_, wavelength_, axis=~0)
-    RGB = sRGB(XYZ, axis=~0)
+    XYZ = XYZcie1931_from_spd(spd_, wavelength_, axis=axis_)
+    RGB = sRGB(XYZ, axis=axis_)
 
     RGB = RGB.to_value(u.dimensionless_unscaled)