From 9ea85acd70a79f99dff9f3a50f3dba3236de84ba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Guillermo=20Sebasti=C3=A1n=20Donatti?= Date: Wed, 18 Sep 2024 13:15:27 +0200 Subject: [PATCH] Correct generator reference to random normal attribute --- keras_cv/src/layers/preprocessing/grid_mask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_cv/src/layers/preprocessing/grid_mask.py b/keras_cv/src/layers/preprocessing/grid_mask.py index 68b303ff1a..3738e976ad 100644 --- a/keras_cv/src/layers/preprocessing/grid_mask.py +++ b/keras_cv/src/layers/preprocessing/grid_mask.py @@ -164,7 +164,7 @@ def get_random_transformation( fill_value = tf.cast(fill_value, dtype=self.compute_dtype) else: # gaussian noise - fill_value = self._random_generator.random_normal( + fill_value = self._random_generator.normal( shape=input_shape, dtype=self.compute_dtype )