A small fix for Python 3.12 compatibility

PiperOrigin-RevId: 705909143
keras-team · Dec 13, 2024 · 6f44991 · 6f44991
1 parent f786954
commit 6f44991
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tf_keras/backend.py b/tf_keras/backend.py
@@ -2029,7 +2029,7 @@ def _create_seed(self, user_specified_seed):
         if user_specified_seed is not None:
             return user_specified_seed
         elif getattr(_SEED_GENERATOR, "generator", None):
-            return _SEED_GENERATOR.generator.randint(1, 1e9)
+            return _SEED_GENERATOR.generator.randint(1, int(1e9))
         else:
             return random.randint(1, int(1e9))
 

diff --git a/tf_keras/utils/tf_utils.py b/tf_keras/utils/tf_utils.py
@@ -78,9 +78,9 @@ def get_random_seed():
       the random seed as an integer.
     """
     if getattr(backend._SEED_GENERATOR, "generator", None):
-        return backend._SEED_GENERATOR.generator.randint(1, 1e9)
+        return backend._SEED_GENERATOR.generator.randint(1, int(1e9))
     else:
-        return random.randint(1, 1e9)
+        return random.randint(1, int(1e9))
 
 
 def is_tensor_or_tensor_list(v):