Obey precommit hook

pymc-devs · Nov 14, 2021 · 488a9c7 · 488a9c7
1 parent 9dc7feb
commit 488a9c7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pymc/sampling_jax.py b/pymc/sampling_jax.py
@@ -168,10 +168,10 @@ def sample_numpyro_nuts(
 
     for i in range(draws):
         for c in range(chains):
-            draw = dict(
-                (value_var.name, raw_samples[c, i])
+            draw = {
+                value_var.name: raw_samples[c, i]
                 for value_var, raw_samples in zip(model.value_vars, raw_mcmc_samples)
-            )
+            }
             sample = fn(draw)
             for vi, v in enumerate(vars_to_sample):
                 mcmc_samples[v.name].append(sample[vi])