Try #8

abhijeetgangan · May 22, 2024 · bec2667 · bec2667
1 parent 3ddd05d
commit bec2667
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/rigid_body_test.py b/tests/rigid_body_test.py
@@ -302,7 +302,7 @@ def total_energy(state, nbrs):
       return (energy_fn(pos, neighbor=nbrs) +
               simulate.kinetic_energy(state))
 
-    nbrs = neighbor_fn.allocate(body)
+    nbrs = neighbor_fn.allocate(body, extra_capacity=2)
     state = init_fn(key, body, 1e-3, mass=shape.mass(), neighbor=nbrs)
     E_initial = total_energy(state, nbrs)