Open GaetanLepage opened 4 months ago
@albcab could you take a look to make this test more robust?
I can't reproduce the failing test on my machine, so it's hard to debug. Making atol=1e-2 and rtol=0 would obviously fix it. Since the harmonic mean is random, increasing chains/steps/learning rates might not necessarily avoid the failing test.
Just for the sake of passing the test in aarch64-linux I would change rtol to atol, @junpenglao thoughts?
Yeah sure.
I can already confirm that the test is fixed thanks to @albcab's patch. Maybe we can wait until the next release to mark this issue as closed.
Things might have changed since then, because several tests fail on 1.2.4.
Skipping test_chees_adaptation
is enough to make the test suite succeed.
Thanks for the feedback, it is a bit difficult for us to replicate, could you paste the full trace for all the test fail? There is not much change on blackjax side re Chees, i suspect it is something upstream in JAX is causing this.
Sure, here it is: https://paste.glepage.com/upload/spider-bee-bison
Describe the issue as clearly as possible:
Using
jax
/jaxlib
0.4.28 andjaxopt
0.8.3, the following test fails onaarch64-linux
:Steps/code to reproduce the bug:
Expected result:
Error message:
Blackjax/JAX/jaxlib/Python version information:
Context for the issue:
No response