Another small fixes in docs and imports (#7030)

* Remove old import from autosummary doc pages

* Fix small error

* Correct place for model_to_networkx & model_to_graphviz

* Correct currentmodule source for pytensorf.rst

* Update docs/source/api/model/core.rst

Co-authored-by: Thomas Wiecki <[email protected]>

* Fix sampling & jax docs imports

---------

Co-authored-by: Denis Kataev <[email protected]>
Co-authored-by: Thomas Wiecki <[email protected]>

Author: 3 people

docs/source/api/math.rst

@@ -67,11 +67,7 @@ Functions exposed in pymc.math
    sgn
    ceil
    floor
-   det
    matrix_inverse
-   extract_diag
-   matrix_dot
-   trace
    sigmoid
    logsumexp
    invlogit

docs/source/api/model/core.rst

@@ -6,8 +6,6 @@ Model creation and inspection
    :toctree: generated/
 
    Model
-   model_to_graphviz
-   model_to_networkx
    modelcontext
 
 Others
@@ -22,3 +20,14 @@ Others
    set_data
    Point
    compile_fn
+
+
+Graph visualization
+-------------------
+
+.. currentmodule:: pymc.model_graph
+.. autosummary::
+   :toctree: generated/
+
+   model_to_networkx
+   model_to_graphviz

docs/source/api/pytensorf.rst

@@ -1,7 +1,7 @@
 PyTensor utils
 **************
 
-.. currentmodule:: pymc
+.. currentmodule:: pymc.pytensorf
 
 .. autosummary::
    :toctree: generated/

docs/source/api/samplers.rst

@@ -4,27 +4,39 @@ Samplers
 This submodule contains functions for MCMC and forward sampling.
 
 
-.. currentmodule:: pymc
+.. currentmodule:: pymc.sampling.forward
 
 .. autosummary::
    :toctree: generated/
 
-   sample
    sample_prior_predictive
    sample_posterior_predictive
-   sample_posterior_predictive_w
-   sampling.jax.sample_blackjax_nuts
-   sampling.jax.sample_numpyro_nuts
-   init_nuts
    draw
 
+
+.. currentmodule:: pymc.sampling.mcmc
+
+.. autosummary::
+   :toctree: generated/
+
+   sample
+   init_nuts
+
+.. currentmodule:: pymc.sampling.jax
+
+.. autosummary::
+   :toctree: generated/
+
+   sample_blackjax_nuts
+   sample_numpyro_nuts
+
+
 Step methods
 ************
 
-.. currentmodule:: pymc
-
 HMC family
 ----------
+.. currentmodule:: pymc.step_methods.hmc
 
 .. autosummary::
    :toctree: generated/
@@ -34,6 +46,7 @@ HMC family
 
 Metropolis family
 -----------------
+.. currentmodule:: pymc.step_methods
 
 .. autosummary::
     :toctree: generated/
@@ -53,6 +66,7 @@ Metropolis family
 
 Other step methods
 ------------------
+.. currentmodule:: pymc.step_methods
 
 .. autosummary::
    :toctree: generated/

pymc/distributions/transforms.py

@@ -57,7 +57,7 @@ def __getattr__(name):
         warnings.warn(f"{name} has been deprecated, use ordered instead.", FutureWarning)
         return ordered
 
-    if name in ("univariate_sum_to_1, multivariate_sum_to_1"):
+    if name in ("univariate_sum_to_1", "multivariate_sum_to_1"):
         warnings.warn(f"{name} has been deprecated, use sum_to_1 instead.", FutureWarning)
         return sum_to_1