From 488a9c7af6edc17024f3ee0446b0c38cf6e64921 Mon Sep 17 00:00:00 2001
From: Rob Zinkov <rob@zinkov.com>
Date: Sun, 14 Nov 2021 19:13:44 +0000
Subject: [PATCH] Obey precommit hook

---
 pymc/sampling_jax.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pymc/sampling_jax.py b/pymc/sampling_jax.py
index 48dde69ad7a..9fce39dbde3 100644
--- 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])