From d4f72c8244b6ca083613e23e7b53326db99db363 Mon Sep 17 00:00:00 2001 From: Wassim Kabalan Date: Tue, 21 Jan 2025 11:21:35 +0100 Subject: [PATCH] format --- notebooks/05-MultiHost_PM.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/notebooks/05-MultiHost_PM.py b/notebooks/05-MultiHost_PM.py index 5be7e60..8174c10 100644 --- a/notebooks/05-MultiHost_PM.py +++ b/notebooks/05-MultiHost_PM.py @@ -105,8 +105,7 @@ def run_simulation(omega_c, sigma8, mesh_shape, box_size, halo_size, halo_size=halo_size, sharding=sharding) - ode_fn = ODETerm( - make_diffrax_ode(mesh_shape, paint_absolute_pos=False)) + ode_fn = ODETerm(make_diffrax_ode(mesh_shape, paint_absolute_pos=False)) # Choose solver solver = LeapfrogMidpoint() if solver_choice == "leapfrog" else Dopri5()