From bec266752505ebf834e9a85607d0861be8c7a5f3 Mon Sep 17 00:00:00 2001 From: abhijeetgangan Date: Wed, 22 May 2024 15:21:08 -0700 Subject: [PATCH] Try #8 --- tests/rigid_body_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/rigid_body_test.py b/tests/rigid_body_test.py index b70d99b..bdfba0d 100644 --- 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)