Inconsistent kwarg progressbar or progress_bar (#6389)

* Changed 'progress_bar' in jax.py to 'progressbar' for consistency; the name 'progress_bar' is used by fastprogress.
* Fixed kwarg in jax.py where MCMC is called back to progress_bar as necessary. The rest of jax.py is still using progressbar over progress_bar for overall project consistency.

Co-authored-by: Michael Osthege <[email protected]>

Author: Gimmesoup

pymc/sampling/jax.py

@@ -506,7 +506,7 @@ def sample_numpyro_nuts(
     initvals: Optional[Union[StartDict, Sequence[Optional[StartDict]]]] = None,
     model: Optional[Model] = None,
     var_names: Optional[Sequence[str]] = None,
-    progress_bar: bool = True,
+    progressbar: bool = True,
     keep_untransformed: bool = False,
     chain_method: str = "parallel",
     postprocessing_backend: Optional[str] = None,
@@ -544,7 +544,7 @@ def sample_numpyro_nuts(
     var_names : sequence of str, optional
         Names of variables for which to compute the posterior samples. Defaults to all
         variables in the posterior.
-    progress_bar : bool, default True
+    progressbar : bool, default True
         Whether or not to display a progress bar in the command line. The bar shows the
         percentage of completion, the sampling speed in samples per second (SPS), and
         the estimated remaining time until completion ("expected time of arrival"; ETA).
@@ -627,7 +627,7 @@ def sample_numpyro_nuts(
         num_chains=chains,
         postprocess_fn=None,
         chain_method=chain_method,
-        progress_bar=progress_bar,
+        progress_bar=progressbar,
     )
 
     tic2 = datetime.now()