From e80768f3dff19d690ab681374a680bf0b29d277a Mon Sep 17 00:00:00 2001 From: dhantule Date: Mon, 14 Jul 2025 14:54:30 +0530 Subject: [PATCH 1/2] fix-searchsorted-type-error --- keras/src/backend/tensorflow/numpy.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/keras/src/backend/tensorflow/numpy.py b/keras/src/backend/tensorflow/numpy.py index b1f9ef211618..b4f87a08f3a0 100644 --- a/keras/src/backend/tensorflow/numpy.py +++ b/keras/src/backend/tensorflow/numpy.py @@ -2134,7 +2134,9 @@ def searchsorted(sorted_sequence, values, side="left"): f"sorted_sequence.shape={sorted_sequence.shape}" ) out_type = ( - "int32" if len(sorted_sequence) <= np.iinfo(np.int32).max else "int64" + "int32" + if (sorted_sequence).shape[0] <= np.iinfo(np.int32).max + else "int64" ) return tf.searchsorted( sorted_sequence, values, side=side, out_type=out_type From a3f9cd75d6607c818f2c61c116fbdaf8a892f0f8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fran=C3=A7ois=20Chollet?= Date: Mon, 14 Jul 2025 09:30:34 -0700 Subject: [PATCH 2/2] Update keras/src/backend/tensorflow/numpy.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- keras/src/backend/tensorflow/numpy.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/keras/src/backend/tensorflow/numpy.py b/keras/src/backend/tensorflow/numpy.py index b4f87a08f3a0..7468ca8db92b 100644 --- a/keras/src/backend/tensorflow/numpy.py +++ b/keras/src/backend/tensorflow/numpy.py @@ -2134,9 +2134,7 @@ def searchsorted(sorted_sequence, values, side="left"): f"sorted_sequence.shape={sorted_sequence.shape}" ) out_type = ( - "int32" - if (sorted_sequence).shape[0] <= np.iinfo(np.int32).max - else "int64" + "int32" if sorted_sequence.shape[0] <= np.iinfo(np.int32).max else "int64" ) return tf.searchsorted( sorted_sequence, values, side=side, out_type=out_type