mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-02-23 10:00:54 +00:00
small fix
This commit is contained in:
parent
d4f72c8244
commit
34bf577b1b
1 changed files with 1 additions and 0 deletions
|
@ -66,6 +66,7 @@ def test_nbody_grad(simulation_config, initial_conditions, lpt_scale_factor,
|
||||||
t1=1.0,
|
t1=1.0,
|
||||||
dt0=None,
|
dt0=None,
|
||||||
y0=y0,
|
y0=y0,
|
||||||
|
args=cosmo,
|
||||||
adjoint=adjoint,
|
adjoint=adjoint,
|
||||||
stepsize_controller=controller,
|
stepsize_controller=controller,
|
||||||
saveat=saveat)
|
saveat=saveat)
|
||||||
|
|
Loading…
Add table
Reference in a new issue