diff --git a/src/augmax/colorspace.py b/src/augmax/colorspace.py index 7272cae..afe15df 100644 --- a/src/augmax/colorspace.py +++ b/src/augmax/colorspace.py @@ -272,7 +272,7 @@ def pixelwise(self, rng: jnp.ndarray, pixel: jnp.ndarray, invert=False) -> jnp.n amount = -amount hue = hue + amount elif op == 'saturation': - F.adjust_brightness(saturation, amount, invert=invert) + saturation = F.adjust_brightness(saturation, amount, invert=invert) transformed = hsv_to_rgb(hue, saturation, value)