Fix to issue #20616

keras/src/ops/numpy.py

@@ -6522,7 +6522,14 @@ def eye(N, M=None, k=0, dtype=None):
 
     Returns:
         Tensor with ones on the k-th diagonal and zeros elsewhere.
-    """
+    Raises:
+        Error if N, M are not integer values.
+    """
+    if not isinstance(N, int):
+        raise ValueError(f"N must be an integer, got {type(N).__name__}")
+    if M is not None and not isinstance(M, int):
+        raise ValueError(f"M must be an integer, got {type(M).__name__}")
+
     return backend.numpy.eye(N, M=M, k=k, dtype=dtype)
 
 

keras/src/ops/numpy_test.py

@@ -4718,6 +4718,8 @@ def test_eye(self):
         # Test k < 0 and M < N and M - k > N
         self.assertAllClose(knp.Eye(k=-2)(4, 3), np.eye(4, 3, k=-2))
 
+
+
     def test_arange(self):
         self.assertAllClose(knp.arange(3), np.arange(3))
         self.assertAllClose(knp.arange(3, 7), np.arange(3, 7))