search

google / objax

Apache License 2.0
768 stars 77 forks source link

Fix tests following jax update #266

Closed GaetanLepage closed 10 months ago

GaetanLepage commented 10 months ago

Some tests were not adapted to the recent changes in the jax API. This PR ensures that all tests pass with the most recent jax.

google-cla[bot] commented 10 months ago

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.

aterzis-google commented 10 months ago

Gaetan thank you for the PR to fix the failing tests!

I just merged a PR that is equivalent so we won't be needing this change.

Thanks again!

GaetanLepage commented 10 months ago

Ok ! Thank you :)