fix: deep copy nuts_sampler_kwarg to prevent .pop side effects (#7652)

inclinedadarsh · web-flow · commit 472da97098d5 · 2025-01-26T19:42:59.000+01:00
diff --git a/pymc/sampling/mcmc.py b/pymc/sampling/mcmc.py
@@ -338,6 +338,7 @@ def _sample_external_nuts(
                 UserWarning,
             )
         compile_kwargs = {}
+        nuts_sampler_kwargs = nuts_sampler_kwargs.copy()
         for kwarg in ("backend", "gradient_backend"):
             if kwarg in nuts_sampler_kwargs:
                 compile_kwargs[kwarg] = nuts_sampler_kwargs.pop(kwarg)

Original file line number	Diff line number	Diff line change
`@@ -338,6 +338,7 @@ def _sample_external_nuts(`
`338`	`338`	`UserWarning,`
`339`	`339`	`)`
`340`	`340`	`compile_kwargs = {}`
	`341`	`+ nuts_sampler_kwargs = nuts_sampler_kwargs.copy()`
`341`	`342`	`for kwarg in ("backend", "gradient_backend"):`
`342`	`343`	`if kwarg in nuts_sampler_kwargs:`
`343`	`344`	`compile_kwargs[kwarg] = nuts_sampler_kwargs.pop(kwarg)`