our feudal pre-commit lord beckons thee

src/dynode/model_odes/seip_model.py

@@ -133,7 +133,9 @@ def compute_ds(strain, immune_state, ds, di_to_w0):
 
     # get all combinations of strain x immune history, jax version of cartesian product
     combinations = jnp.stack(
-        jnp.meshgrid(jnp.arange(p.NUM_STRAINS), jnp.arange(2**p.NUM_STRAINS)),
+        jnp.meshgrid(
+            jnp.arange(p.NUM_STRAINS), jnp.arange(2**p.NUM_STRAINS)
+        ),
         axis=-1,
     ).reshape(-1, 2)
     # compute vectorized function on all possible immune_hist x exposing strain