Remove experimental warning from external nuts samplers. (#6887)

twiecki · web-flow · commit dfb05b6893fe · 2023-09-01T12:44:48.000+02:00
diff --git a/pymc/sampling/mcmc.py b/pymc/sampling/mcmc.py
@@ -270,8 +270,6 @@ def _sample_external_nuts(
     nuts_sampler_kwargs: Optional[Dict],
     **kwargs,
 ):
-    warnings.warn("Use of external NUTS sampler is still experimental", UserWarning)
-
     if nuts_sampler_kwargs is None:
         nuts_sampler_kwargs = {}
 
diff --git a/tests/sampling/test_mcmc_external.py b/tests/sampling/test_mcmc_external.py
@@ -46,8 +46,6 @@ def test_external_nuts_sampler(recwarn, nuts_sampler):
         if warn.category is not FutureWarning
     }
     expected = set()
-    if nuts_sampler != "pymc":
-        expected.add((UserWarning, "Use of external NUTS sampler is still experimental"))
     if nuts_sampler == "nutpie":
         expected.add(
             (

Original file line number	Diff line number	Diff line change
`@@ -46,8 +46,6 @@ def test_external_nuts_sampler(recwarn, nuts_sampler):`
`46`	`46`	`if warn.category is not FutureWarning`
`47`	`47`	`}`
`48`	`48`	`expected = set()`
`49`		`- if nuts_sampler != "pymc":`
`50`		`- expected.add((UserWarning, "Use of external NUTS sampler is still experimental"))`
`51`	`49`	`if nuts_sampler == "nutpie":`
`52`	`50`	`expected.add(`
`53`	`51`	`(`