Ensure keras.ops.eye behavior is consistent across backends. (#21738)

* ensure eye behavior is consistent across backends

* Update keras/src/ops/numpy_test.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

* simplify per pr review

* pre-commit

* fix test for torch backend + add comments

* update implementation to raise TypeError for consistency

* add case for M being the onl float

* improve naming of inner function for type check

---------

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: danielenricocahall

keras/src/ops/numpy.py

@@ -7228,6 +7228,19 @@ def eye(N, M=None, k=0, dtype=None):
     Returns:
         Tensor with ones on the k-th diagonal and zeros elsewhere.
     """
+
+    def is_floating_type(v):
+        return (
+            isinstance(v, float)
+            or getattr(v, "dtype", None) in dtypes.FLOAT_TYPES
+        )
+
+    if is_floating_type(N):
+        raise TypeError("Argument `N` must be an integer or an integer tensor.")
+    if is_floating_type(M):
+        raise TypeError(
+            "Argument `M` must be an integer, an integer tensor, or `None`."
+        )
     return backend.numpy.eye(N, M=M, k=k, dtype=dtype)
 
 

keras/src/ops/numpy_test.py

@@ -5230,6 +5230,19 @@ def test_eye(self):
         # Test k < 0 and M < N and M - k > N
         self.assertAllClose(knp.eye(4, 3, k=-2), np.eye(4, 3, k=-2))
 
+    def test_eye_raises_error_with_floats(self):
+        with self.assertRaises(TypeError):
+            knp.eye(3.0)
+        with self.assertRaises(TypeError):
+            knp.eye(3.0, 2.0)
+        with self.assertRaises(TypeError):
+            knp.eye(3, 2.0)
+        with self.assertRaises(TypeError):
+            v = knp.max(knp.arange(4.0))
+            knp.eye(v)
+        with self.assertRaises(TypeError):
+            knp.eye(knp.array(3, dtype="bfloat16"))
+
     def test_arange(self):
         self.assertAllClose(knp.arange(3), np.arange(3))
         self.assertAllClose(knp.arange(3, 7), np.arange(3, 7))