improved concision

pyro-ppl · Nov 9, 2023 · e52be4d · e52be4d
1 parent a2d54ba
commit e52be4d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/numpyro/infer/hmc_gibbs.py b/numpyro/infer/hmc_gibbs.py
@@ -137,8 +137,7 @@ def init(self, rng_key, num_warmup, init_params, model_args, model_kwargs):
 
         for name, site in self._prototype_trace.items():
             if init_params and name in init_params:
-                gibbs_sites[name] = init_params[name]
-                init_params.pop(name)
+                gibbs_sites[name] = init_params.pop(name)
 
             elif name in self._gibbs_sites:
                 gibbs_sites[name] = site["value"]