Expose aesara.tensor.random.basic via aesara.tensor.random

aesara/tensor/__init__.py

@@ -253,3 +253,6 @@ from aesara.tensor.type import (
     zvector,
 )
 from aesara.tensor.type_other import *
+
+
+__all__ = ["random"]  # noqa: F405

aesara/tensor/random/__init__.py

 # Initialize `RandomVariable` optimizations
 import aesara.tensor.random.opt
 import aesara.tensor.random.utils
+from aesara.tensor.random.basic import *
+from aesara.tensor.random.utils import RandomStream

aesara/tensor/random/basic.py

@@ -693,3 +693,41 @@ class PermutationRV(RandomVariable):
 
 
 permutation = PermutationRV()
+
+
+__all__ = [
+    "permutation",
+    "choice",
+    "randint",
+    "categorical",
+    "multinomial",
+    "betabinom",
+    "nbinom",
+    "binomial",
+    "laplace",
+    "bernoulli",
+    "truncexpon",
+    "wald",
+    "invgamma",
+    "halfcauchy",
+    "cauchy",
+    "hypergeometric",
+    "geometric",
+    "poisson",
+    "dirichlet",
+    "multivariate_normal",
+    "vonmises",
+    "logistic",
+    "weibull",
+    "exponential",
+    "gumbel",
+    "pareto",
+    "chisquare",
+    "gamma",
+    "lognormal",
+    "halfnormal",
+    "normal",
+    "beta",
+    "triangular",
+    "uniform",
+]