removed thinning, updated docstrings and refs

Author: kylejcaron

external_tests/helpers.py

@@ -132,7 +132,7 @@ def _numpyro_noncentered_guide(J, sigma, y=None):
 
 
 def numpyro_schools_model(data, draws, chains):
-    """Noncentered eight schools implementation in NumPyro."""
+    """Non-centered eight schools implementation in NumPyro."""
     from jax.random import PRNGKey
     from numpyro.infer import MCMC, NUTS
 
@@ -156,7 +156,7 @@ def numpyro_schools_model(data, draws, chains):
 
 
 def numpyro_schools_model_svi(data, draws, chains):
-    """Centered eight schools implementation in NumPyro."""
+    """Non-centered eight schools implementation in NumPyro."""
     from jax.random import PRNGKey
     from numpyro.infer import SVI, Trace_ELBO, init_to_sample
     from numpyro.infer.autoguide import AutoNormal
@@ -169,7 +169,7 @@ def numpyro_schools_model_svi(data, draws, chains):
 
 
 def numpyro_schools_model_svi_custom_guide(data, draws, chains):
-    """Centered eight schools implementation in NumPyro."""
+    """Non-centered eight schools implementation in NumPyro."""
     from jax.random import PRNGKey
     from numpyro.infer import SVI, Trace_ELBO
     from numpyro.optim import Adam

src/arviz_base/io_numpyro.py

@@ -22,7 +22,6 @@ def __init__(
         model_args=None,
         model_kwargs=None,
         num_samples: int = 1000,
-        thinning: int = 1,
     ):
         import jax
         import numpyro
@@ -32,7 +31,7 @@ def __init__(
         self._args = model_args or tuple()
         self._kwargs = model_kwargs or dict()
         self.num_samples = num_samples
-        self.thinning = thinning
+        self.thinning = 1
         self.num_chains = 0
         self.sample_dims = ["samples"]
         self.kind = "svi"

src/arviz_base/io_numpyro.pyi

@@ -24,7 +24,6 @@ class SVIWrapper:
         model_args=...,
         model_kwargs=...,
         num_samples: int = ...,
-        thinning: int = ...,
     ) -> None: ...
     def get_samples(self, seed=..., **kwargs) -> None: ...
     @property