Fix CI I guess

keras-team · Dec 11, 2024 · e12ae46 · e12ae46
1 parent 95b4383
commit e12ae46
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras/src/backend/tensorflow/random.py b/keras/src/backend/tensorflow/random.py
@@ -94,6 +94,7 @@ def dropout(inputs, rate, noise_shape=None, seed=None):
 
 
 def shuffle(x, axis=0, seed=None):
+    seed = _cast_seed(draw_seed(seed))
     indices = tf.argsort(
         tf.random.stateless_uniform(shape=[tf.shape(x)[axis]], seed=seed)
     )