regenerate stub files after rebase

Author: kylejcaron

src/arviz_base/__init__.pyi

@@ -25,7 +25,7 @@ from arviz_base.datasets import (
 from arviz_base.io_cmdstanpy import from_cmdstanpy
 from arviz_base.io_dict import from_dict
 from arviz_base.io_emcee import from_emcee
-from arviz_base.io_numpyro import from_numpyro
+from arviz_base.io_numpyro import from_numpyro, from_numpyro_svi
 from arviz_base.io_pystan import from_pystan
 from arviz_base.rcparams import rc_context, rcParams
 from arviz_base.reorg import (
@@ -55,6 +55,7 @@ __all__ = [
     "from_dict",
     "from_emcee",
     "from_numpyro",
+    "from_numpyro_svi",
     "rc_context",
     "rcParams",
     "extract",

src/arviz_base/io_numpyro.pyi

@@ -14,6 +14,21 @@ from arviz_base.base import dict_to_dataset, requires
 from arviz_base.rcparams import rc_context, rcParams
 from arviz_base.utils import expand_dims
 
+class SVIWrapper:
+    def __init__(
+        self,
+        svi,
+        *,
+        svi_result,
+        model_args=...,
+        model_kwargs=...,
+        num_samples: int = ...,
+    ) -> None: ...
+    def get_samples(self, seed=..., **kwargs) -> None: ...
+    @property
+    def sampler(self) -> None: ...
+    def get_extra_fields(self, **kwargs) -> None: ...
+
 def _add_dims(
     dims_a: dict[str, list[str]], dims_b: dict[str, list[str]]
 ) -> dict[str, list[str]]: ...
@@ -46,7 +61,7 @@ class NumPyroConverter:
         extra_event_dims: dict | None = ...,
         num_chains: int = ...,
     ) -> None: ...
-    def _get_model_trace(self, model, args, kwargs, key) -> None: ...
+    def _get_model_trace(self, model, model_args, model_kwargs, key) -> None: ...
     def posterior_to_xarray(self) -> None: ...
     def sample_stats_to_xarray(self) -> None: ...
     def log_likelihood_to_xarray(self) -> None: ...
@@ -77,3 +92,23 @@ def from_numpyro(
     extra_event_dims: dict | None = ...,
     num_chains: int = ...,
 ) -> DataTree: ...
+def from_numpyro_svi(
+    svi: numpyro.infer.svi.SVI,
+    *,
+    svi_result: numpyro.infer.svi.SVIRunResult,
+    model_args: tuple | None = ...,
+    model_kwargs: dict | None = ...,
+    prior: dict | None = ...,
+    posterior_predictive: dict | None = ...,
+    predictions: dict | None = ...,
+    constant_data: dict | None = ...,
+    predictions_constant_data: dict | None = ...,
+    log_likelihood=...,
+    index_origin: int | None = ...,
+    coords: dict | None = ...,
+    dims: dict[str, list[str]] | None = ...,
+    pred_dims: dict | None = ...,
+    extra_event_dims: dict | None = ...,
+    model=...,
+    num_samples: int = ...,
+) -> DataTree: ...