Adjust test tolerances for test_assert_trees_all_close #334
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Adjust test tolerances for test_assert_trees_all_close
The test for chexify
assert_trees_all_close
asserted thatjnp.isclose(10, 20, rtol=0.5, atol0.5)
is false. By the rules ofnp.testing.assert_allclose
thisshould actually pass, because (|10-20| <= 0.5 + (0.5 * 20)). However, it was only
when JAX PR #19936 was merged that this became the case.
This changes the rtol/atol to make the test pass.