Fix seed

zaxtax · zaxtax · commit 59ebfdb22286 · 2021-11-17T16:00:21.000Z
diff --git a/pymc/sampling_jax.py b/pymc/sampling_jax.py
@@ -174,6 +174,7 @@ def sample_numpyro_nuts(
 
     if chains == 1:
         init_params=init_state
+        map_seed = seed
     else:
         init_params=init_state_batched
 
