Skip to content

Conversation

danielenricocahall
Copy link
Contributor

This PR is addressing #20616 by adding a check to the eye function to ensure that N and M are either integers or Tensors with an integer dtype, which is consistent with the current numpy behavior.

Copy link

google-cla bot commented Oct 13, 2025

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

Copy link
Contributor

Summary of Changes

Hello @danielenricocahall, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request addresses an inconsistency in the keras.ops.eye function by enforcing strict type checking for its N and M arguments. By ensuring these parameters are always integers or integer tensors, the change standardizes the function's behavior to match numpy's expectations, thereby preventing potential runtime errors and improving the overall robustness and predictability of Keras operations across different backends.

Highlights

  • Input Validation for keras.ops.eye: Introduced validation checks within the keras.ops.eye function to ensure that the N and M parameters are strictly integers or integer tensors, aligning its behavior with numpy.
  • Error Handling for Float Inputs: Added ValueError exceptions that are raised when N or M are provided as float values, preventing incorrect usage and promoting type consistency.
  • Enhanced Test Coverage: Included new test cases in numpy_test.py specifically designed to verify that keras.ops.eye correctly raises ValueError when float inputs are passed for N or M.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request enhances the keras.ops.eye function by adding checks to ensure its arguments are integers or integer tensors, aligning its behavior with NumPy and improving consistency across backends. The changes are well-implemented and include corresponding tests. I've provided a couple of suggestions to further improve the code's robustness and test coverage.

Comment on lines 7231 to 7233
def is_float(v):
if isinstance(v, float) or getattr(v, "dtype", None) in ("float16", "float32", "float64"):
return True
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

The implementation of is_float can be simplified to be more Pythonic. Also, it's missing bfloat16 from the check. Using the dtypes.FLOAT_TYPES constant would make this more robust and readable.

    def is_float(v):
        return isinstance(v, float) or getattr(v, "dtype", None) in dtypes.FLOAT_TYPES

@codecov-commenter
Copy link

codecov-commenter commented Oct 13, 2025

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 82.59%. Comparing base (3137cb0) to head (392d05c).

Additional details and impacted files
@@           Coverage Diff           @@
##           master   #21738   +/-   ##
=======================================
  Coverage   82.59%   82.59%           
=======================================
  Files         572      572           
  Lines       58535    58541    +6     
  Branches     9158     9160    +2     
=======================================
+ Hits        48345    48351    +6     
  Misses       7853     7853           
  Partials     2337     2337           
Flag Coverage Δ
keras 82.39% <100.00%> (+<0.01%) ⬆️
keras-jax 63.20% <100.00%> (+<0.01%) ⬆️
keras-numpy 57.56% <100.00%> (+<0.01%) ⬆️
keras-openvino 34.33% <0.00%> (-0.01%) ⬇️
keras-tensorflow 63.95% <100.00%> (+<0.01%) ⬆️
keras-torch 63.49% <100.00%> (+<0.01%) ⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

Copy link
Collaborator

@hertschuh hertschuh left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the PR!

@google-ml-butler google-ml-butler bot added kokoro:force-run ready to pull Ready to be merged into the codebase labels Oct 14, 2025
@hertschuh hertschuh merged commit 14144cb into keras-team:master Oct 14, 2025
11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready to pull Ready to be merged into the codebase size:S

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants