Fixes "'float' object cannot be interpreted as an integer" in Python …

…3.12 PiperOrigin-RevId: 718392316
keras-team · Jan 27, 2025 · 61647ab · 61647ab
1 parent ad06941
commit 61647ab
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tf_keras/preprocessing/sequence.py b/tf_keras/preprocessing/sequence.py
@@ -365,7 +365,7 @@ def skipgrams(
         random.shuffle(words)
 
         couples += [
-            [words[i % len(words)], random.randint(1, vocabulary_size - 1)]
+            [words[i % len(words)], random.randint(1, int(vocabulary_size - 1))]
             for i in range(num_negative_samples)
         ]
         if categorical:
@@ -375,7 +375,7 @@ def skipgrams(
 
     if shuffle:
         if seed is None:
-            seed = random.randint(0, 10e6)
+            seed = random.randint(0, int(10e6))
         random.seed(seed)
         random.shuffle(couples)
         random.seed(seed)

diff --git a/tf_keras/utils/text_dataset_test.py b/tf_keras/utils/text_dataset_test.py
@@ -33,7 +33,7 @@ def _prepare_directory(
     ):
         # Get a unique temp directory
         temp_dir = os.path.join(
-            self.get_temp_dir(), str(random.randint(0, 1e6))
+            self.get_temp_dir(), str(random.randint(0, int(1e6)))
         )
         os.mkdir(temp_dir)
         self.addCleanup(shutil.rmtree, temp_dir)