Fix random_saturation

keras-team · Dec 17, 2024 · c0a1f24 · c0a1f24
1 parent 9bcf324
commit c0a1f24
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/keras/src/layers/preprocessing/image_preprocessing/random_saturation.py b/keras/src/layers/preprocessing/image_preprocessing/random_saturation.py
@@ -99,7 +99,7 @@ def get_random_transformation(self, data, training=True, seed=None):
         return {"factor": factor}
 
     def transform_images(self, images, transformation=None, training=True):
-        def _apply_random_saturation(images, transformation):
+        if training:
             adjust_factors = transformation["factor"]
             adjust_factors = self.backend.cast(
                 adjust_factors, self.compute_dtype
@@ -133,10 +133,6 @@ def _apply_random_saturation(images, transformation):
             images = self.backend.image.hsv_to_rgb(
                 images, data_format=self.data_format
             )
-            return images
-
-        if training:
-            images = _apply_random_saturation(images, transformation)
         return images
 
     def transform_labels(self, labels, transformation, training=True):