From f74293dea8d76ec29cd6b213e2daf3ce59c36962 Mon Sep 17 00:00:00 2001 From: Haohuan Wang Date: Wed, 6 Aug 2025 13:15:46 -0700 Subject: [PATCH] update saturation --- src/augmax/colorspace.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)