Correct failed test case

keras-team · Dec 21, 2024 · 6b25c98 · 6b25c98
1 parent c366d3e
commit 6b25c98
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/keras/src/layers/preprocessing/image_preprocessing/random_color_jitter.py b/keras/src/layers/preprocessing/image_preprocessing/random_color_jitter.py
@@ -114,6 +114,9 @@ def __init__(
 
     def transform_images(self, images, transformation, training=True):
         if training:
+            if backend_utils.in_tf_graph():
+                self.backend.set_backend("tensorflow")
+            images = self.backend.cast(images, self.compute_dtype)
             if self.brightness_factor is not None:
                 if backend_utils.in_tf_graph():
                     self.random_brightness.backend.set_backend("tensorflow")
@@ -132,6 +135,9 @@ def transform_images(self, images, transformation, training=True):
                 transformation = self.random_contrast.get_random_transformation(
                     images, seed=self._get_seed_generator(self.backend._backend)
                 )
+                transformation["contrast_factor"] = self.backend.cast(
+                    transformation["contrast_factor"], dtype=self.compute_dtype
+                )
                 images = self.random_contrast.transform_images(
                     images, transformation
                 )