small fix

2025-02-22 17:47:11 +00:00 · 2025-01-21 12:08:47 +01:00 · 2025-01-21 12:08:47 +01:00 · 34bf577b1b
commit 34bf577b1b
parent d4f72c8244
1 changed files with 1 additions and 0 deletions
--- a/tests/test_gradients.py
+++ b/tests/test_gradients.py
@ -66,6 +66,7 @@ def test_nbody_grad(simulation_config, initial_conditions, lpt_scale_factor,
                                t1=1.0,
                                dt0=None,
                                y0=y0,
+                                args=cosmo,
                                adjoint=adjoint,
                                stepsize_controller=controller,
                                saveat=saveat)