Fix floating-point precision in test_b_jax.py

Corrected formatting of floating-point numbers in test data.

Author: heptaflar

test/b/test_b_jax.py

@@ -180,7 +180,7 @@ def test_b_poly_grid(self):
                     [17.1208, 17.0362, 17.6470],
                 ],
                 [
-                    [34.5286, 34.444, 35.05480],
+                    [34.5286, 34.4440, 35.0548],
                     [32.1607, 32.0761, 32.6869],
                     [31.7800, 31.6954, 32.3062],
                 ],