From b949827e92e288e0a15556cc1dbb2945ce5a8c5a Mon Sep 17 00:00:00 2001 From: Francois Lanusse Date: Fri, 19 Jul 2024 10:49:51 -0400 Subject: [PATCH] Update jaxpm/pm.py --- jaxpm/pm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jaxpm/pm.py b/jaxpm/pm.py index 686cc07..9b14a87 100644 --- a/jaxpm/pm.py +++ b/jaxpm/pm.py @@ -83,7 +83,7 @@ def make_ode_fn(mesh_shape): return nbody_ode -def pgd_correction(pos, mesh_shape, cosmo, params): +def pgd_correction(pos, mesh_shape, params): """ improve the short-range interactions of PM-Nbody simulations with potential gradient descent method, based on https://arxiv.org/abs/1804.00671 args: