From 1f2035176f49078ceda3e44f8b52a51e98fed075 Mon Sep 17 00:00:00 2001 From: Wassim KABALAN Date: Fri, 19 Jul 2024 19:25:26 +0200 Subject: [PATCH] Update scripts/distributed_pm.py Co-authored-by: Francois Lanusse --- scripts/distributed_pm.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/scripts/distributed_pm.py b/scripts/distributed_pm.py index 1818eea..5f41af4 100644 --- a/scripts/distributed_pm.py +++ b/scripts/distributed_pm.py @@ -47,9 +47,6 @@ def run_simulation(omega_c, sigma8): pk_fn, seed=jax.random.PRNGKey(0)) - # Create particles - particles = jnp.stack(jnp.meshgrid(*[jnp.arange(s) for s in mesh_shape]), - axis=-1).reshape([-1, 3]) cosmo = jc.Planck15(Omega_c=omega_c, sigma8=sigma8)