Skip to content

Commit 765fdef

Browse files
authored
Removed **kwargs from sample_numpyro_nuts and sample_blackjax_nuts (#6768)
1 parent 864ecb3 commit 765fdef

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

pymc/sampling/jax.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,6 @@ def sample_blackjax_nuts(
317317
postprocessing_backend: Optional[str] = None,
318318
postprocessing_chunks: Optional[int] = None,
319319
idata_kwargs: Optional[Dict[str, Any]] = None,
320-
**kwargs,
321320
) -> az.InferenceData:
322321
"""
323322
Draw samples from the posterior using the NUTS method from the ``blackjax`` library.
@@ -530,7 +529,6 @@ def sample_numpyro_nuts(
530529
postprocessing_chunks: Optional[int] = None,
531530
idata_kwargs: Optional[Dict] = None,
532531
nuts_kwargs: Optional[Dict] = None,
533-
**kwargs,
534532
) -> az.InferenceData:
535533
"""
536534
Draw samples from the posterior using the NUTS method from the ``numpyro`` library.

0 commit comments

Comments
 (0)