search

google / objax

Apache License 2.0
773 stars 77 forks source link

Fix tests following jax update #266

Closed GaetanLepage closed 1 year ago

GaetanLepage commented 1 year 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 1 year 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 1 year 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 1 year ago

Ok ! Thank you :)