Revert removal of pm.compute_log_likelihood

Author: jessegrabowski

docs/source/api/misc.rst

@@ -6,6 +6,7 @@ Other utils
 .. autosummary::
    :toctree: generated/
 
+   compute_log_likelihood
    find_constrained_prior
 
 .. currentmodule:: pymc.blocking

docs/source/learn/core_notebooks/model_comparison.ipynb

@@ -260,7 +260,7 @@
     "\n",
     "### Model log-likelihood\n",
     "\n",
-    "In order to compute LOO, ArviZ needs access to the model elemwise loglikelihood for every posterior sample. We can add it via {func}`~pymc.stats.compute_log_likelihood`. Alternatively we can pass `idata_kwargs={\"log_likelihood\": True}` to {func}`~pymc.sample` to have it computed automatically at the end of sampling."
+    "In order to compute LOO, ArviZ needs access to the model elemwise loglikelihood for every posterior sample. We can add it via {func}`~pymc.compute_log_likelihood`. Alternatively we can pass `idata_kwargs={\"log_likelihood\": True}` to {func}`~pymc.sample` to have it computed automatically at the end of sampling."
    ]
   },
   {
@@ -295,7 +295,7 @@
    ],
    "source": [
     "with pooled:\n",
-    "    pm.stats.compute_log_likelihood(trace_p)"
+    "    pm.compute_log_likelihood(trace_p)"
    ]
   },
   {
@@ -363,7 +363,7 @@
    ],
    "source": [
     "with hierarchical:\n",
-    "    pm.stats.compute_log_likelihood(trace_h)"
+    "    pm.compute_log_likelihood(trace_h)"
    ]
   },
   {

pymc/__init__.py

@@ -59,6 +59,7 @@ def __set_compiler_flags():
 from pymc.pytensorf import compile
 from pymc.sampling import *
 from pymc.smc import *
+from pymc.stats import compute_log_likelihood
 from pymc.step_methods import *
 from pymc.tuning import *
 from pymc.variational import *

pymc/sampling/jax.py

@@ -652,7 +652,7 @@ def sample_jax_nuts(
     if idata_kwargs.pop("log_likelihood", False):
         warnings.warn(
             "`passing log_likelihood` is deprecated and will be removed in future versions. Use "
-            ":func:`pymc.stats.compute_log_likelihood` instead.",
+            ":func:`pymc.compute_log_likelihood` instead.",
             FutureWarning,
             stacklevel=3,
         )

pymc/sampling/mcmc.py

@@ -1095,7 +1095,7 @@ def _sample_return(
             if log_likelihood:
                 warnings.warn(
                     "`passing log_likelihood` is deprecated and will be removed in future versions. Use "
-                    ":func:`pymc.stats.compute_log_likelihood` instead.",
+                    ":func:`pymc.compute_log_likelihood` instead.",
                     FutureWarning,
                     stacklevel=2,
                 )