keras-team / keras-core

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
Apache License 2.0
1.27k stars 115 forks source link

Adds JAX version checking for Export #923

Closed nkovela1 closed 10 months ago

nkovela1 commented 10 months ago

This PR is a temporary fix in response to the incompatibility between the latest JAX release and latest stable TF release, affecting the export API and its use of JAX2TF.

codecov[bot] commented 10 months ago

Codecov Report

Patch coverage has no change and project coverage change: -0.02% :warning:

Comparison is base (c64de55) 79.73% compared to head (0e7cbb9) 79.72%.

Additional details and impacted files ```diff @@ Coverage Diff @@ ## main #923 +/- ## ========================================== - Coverage 79.73% 79.72% -0.02% ========================================== Files 318 318 Lines 28627 28631 +4 Branches 5447 5449 +2 ========================================== Hits 22827 22827 - Misses 4333 4336 +3 - Partials 1467 1468 +1 ``` | [Flag](https://app.codecov.io/gh/keras-team/keras-core/pull/923/flags?src=pr&el=flags&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team) | Coverage Δ | | |---|---|---| | [keras_core](https://app.codecov.io/gh/keras-team/keras-core/pull/923/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team) | `79.65% <0.00%> (-0.02%)` | :arrow_down: | | [keras_core-numpy](https://app.codecov.io/gh/keras-team/keras-core/pull/923/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team) | `60.38% <0.00%> (-0.01%)` | :arrow_down: | | [keras_core-tensorflow](https://app.codecov.io/gh/keras-team/keras-core/pull/923/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team) | `66.81% <0.00%> (-0.01%)` | :arrow_down: | | [keras_core-torch](https://app.codecov.io/gh/keras-team/keras-core/pull/923/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team) | `69.20% <0.00%> (-0.01%)` | :arrow_down: | Flags with carried forward coverage won't be shown. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team#carryforward-flags-in-the-pull-request-comment) to find out more. | [Files Changed](https://app.codecov.io/gh/keras-team/keras-core/pull/923?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team) | Coverage Δ | | |---|---|---| | [keras\_core/export/export\_lib.py](https://app.codecov.io/gh/keras-team/keras-core/pull/923?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team#diff-a2VyYXNfY29yZS9leHBvcnQvZXhwb3J0X2xpYi5weQ==) | `77.72% <0.00%> (-1.51%)` | :arrow_down: |

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.