diff --git a/keras_hub/src/layers/preprocessing/random_deletion.py b/keras_hub/src/layers/preprocessing/random_deletion.py index ed4411e230..f0a1e6342f 100644 --- a/keras_hub/src/layers/preprocessing/random_deletion.py +++ b/keras_hub/src/layers/preprocessing/random_deletion.py @@ -125,7 +125,7 @@ def __init__( self.rate = rate self.max_deletions = max_deletions - self.seed = random.randint(1, 1e9) if seed is None else seed + self.seed = random.randint(1, int(1e9)) if seed is None else seed self._generator = tf.random.Generator.from_seed(self.seed) self.skip_list = skip_list self.skip_fn = skip_fn diff --git a/keras_hub/src/layers/preprocessing/random_swap.py b/keras_hub/src/layers/preprocessing/random_swap.py index 2884ce0e36..624b3e6663 100644 --- a/keras_hub/src/layers/preprocessing/random_swap.py +++ b/keras_hub/src/layers/preprocessing/random_swap.py @@ -127,7 +127,7 @@ def __init__( self.rate = rate self.max_swaps = max_swaps - self.seed = random.randint(1, 1e9) if seed is None else seed + self.seed = random.randint(1, int(1e9)) if seed is None else seed self._generator = tf.random.Generator.from_seed(self.seed) self.skip_list = skip_list self.skip_fn = skip_fn