document sampling related functions as users should import them

pymc-devs · Apr 6, 2024 · 9fd90c0 · 9fd90c0
1 parent 3d3a4fd
commit 9fd90c0
Showing 1 changed file with 4 additions and 19 deletions.
diff --git a/docs/source/api/samplers.rst b/docs/source/api/samplers.rst
@@ -4,34 +4,19 @@ Samplers
 This submodule contains functions for MCMC and forward sampling.
 
 
-.. currentmodule:: pymc.sampling.forward
+.. currentmodule:: pymc
 
 .. autosummary::
    :toctree: generated/
 
+   sample
    sample_prior_predictive
    sample_posterior_predictive
    draw
-
-.. currentmodule:: pymc.sampling.deterministic
    compute_deterministics
-
-
-.. currentmodule:: pymc.sampling.mcmc
-
-.. autosummary::
-   :toctree: generated/
-
-   sample
    init_nuts
-
-.. currentmodule:: pymc.sampling.jax
-
-.. autosummary::
-   :toctree: generated/
-
-   sample_blackjax_nuts
-   sample_numpyro_nuts
+   sampling.jax.sample_blackjax_nuts
+   sampling.jax.sample_numpyro_nuts
 
 
 Step methods