You are working on the jax-ml/jax repository. The source code is checked out at /testbed at commit…

You are working on the jax-ml/jax repository. The source code is checked out at /testbed at commit…: a task in SWE-Lego-RL-2699 (Harbor dataset). nan location mismatch: ACTUAL: array( nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, inf, 1.000e+00, 1.000e+00, 2.000e+00, 6.000e+00, 2.400e+01…

The task

nan location mismatch: ACTUAL: array([ nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, inf, 1.000e+00, 1.000e+00, 2.000e+00, 6.000e+00, 2.400e+01, 1.200e+02, 7.200e+02, 5.040e+03, 4.032e+04]) DESIRED: array([ inf, inf, inf, inf, inf, inf, inf, inf, inf, inf, inf, 1.000e+00, 1.000e+00, 2.000e+00, 6.000e+00,…

Part of Lego-X/Lego-RL-2699.