Fix warning for random_saturation (#21066)

shashaka · fchollet · web-flow · commit 9fea22aaca4a · 2025-03-20T22:26:07.000-07:00
* Fix warning for random_saturation

* Update random_saturation.py

* Update random_saturation.py

* Update 1e-6 to epsilon()

* merge master

---------

Co-authored-by: François Chollet &lt;francois.chollet@gmail.com&gt;
diff --git a/keras/src/layers/preprocessing/image_preprocessing/random_saturation.py b/keras/src/layers/preprocessing/image_preprocessing/random_saturation.py
@@ -1,4 +1,5 @@
 from keras.src.api_export import keras_export
+from keras.src.backend import epsilon
 from keras.src.layers.preprocessing.image_preprocessing.base_image_preprocessing_layer import (  # noqa: E501
     BaseImagePreprocessingLayer,
 )
@@ -95,7 +96,7 @@ def get_random_transformation(self, data, training=True, seed=None):
             maxval=self.factor[1],
             seed=seed,
         )
-        factor = factor / (1 - factor)
+        factor = factor / (1 - factor + epsilon())
         return {"factor": factor}
 
     def transform_images(self, images, transformation=None, training=True):

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`from keras.src.api_export import keras_export`
	`2`	`+from keras.src.backend import epsilon`
`2`	`3`	`from keras.src.layers.preprocessing.image_preprocessing.base_image_preprocessing_layer import ( # noqa: E501`
`3`	`4`	`BaseImagePreprocessingLayer,`
`4`	`5`	`)`
`@@ -95,7 +96,7 @@ def get_random_transformation(self, data, training=True, seed=None):`
`95`	`96`	`maxval=self.factor[1],`
`96`	`97`	`seed=seed,`
`97`	`98`	`)`
`98`		`- factor = factor / (1 - factor)`
	`99`	`+ factor = factor / (1 - factor + epsilon())`
`99`	`100`	`return {"factor": factor}`
`100`	`101`
`101`	`102`	`def transform_images(self, images, transformation=None, training=True):`