Skip to content

Commit dd5908e

Browse files
add check for np.integer
1 parent 315ed9b commit dd5908e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

keras/src/backend/openvino/numpy.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1329,7 +1329,9 @@ def repeat(x, repeats, axis=None):
13291329
and repeats.size == 1
13301330
):
13311331
repeats_val = (
1332-
int(repeats) if isinstance(repeats, np.ndarray) else repeats
1332+
int(repeats)
1333+
if isinstance(repeats, (np.integer, np.ndarray))
1334+
else repeats
13331335
)
13341336
dim_len = ov_opset.gather(
13351337
ov_opset.shape_of(x, Type.i32),

0 commit comments

Comments
 (0)