From 81f06d1d3a23a1ecb48163466857d1b175a11a98 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 27 Jan 2025 10:53:34 -0800 Subject: [PATCH] Fixes "'float' object cannot be interpreted as an integer" in Python 3.12 PiperOrigin-RevId: 720231366 --- tf_keras/preprocessing/sequence.py | 4 ++-- tf_keras/utils/text_dataset_test.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tf_keras/preprocessing/sequence.py b/tf_keras/preprocessing/sequence.py index 291c7638e..e935322c2 100644 --- 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 index 50349618b..df4419899 100644 --- 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)