codecov[bot]
commented
1 year ago Codecov Report
:exclamation: No coverage uploaded for pull request base (
main@1a7720e). Click here to learn what that means. Patch has no changes to coverable lines.
Additional details and impacted files
```diff @@ Coverage Diff @@ ## main #819 +/- ## ======================================= Coverage ? 76.01% ======================================= Files ? 328 Lines ? 31134 Branches ? 6060 ======================================= Hits ? 23668 Misses ? 5868 Partials ? 1598 ``` | [Flag](https://app.codecov.io/gh/keras-team/keras-core/pull/819/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/819/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=keras-team) | `75.92% <0.00%> (?)` | | 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.:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
This PR adds initial support for the export of JAX-backend models to TF SavedModel. Additionally, a test has been added for subclassed model export support along with logic in both TF and JAX backends.
There are a few caveats and TODOs to note here:
jax2tf function conversion is lacking argument name preservation, causing the names of arguments passed to become
args_tf_0,args_tf_1, etc. I have opened a bug internally in JAX2TF Users regarding this, and a workaround should be included in a subsequent PR.ReloadedLayer support will be included in a subsequent PR, further discussion will be needed on this.
All other tests pass!