Add NumPy random function alias to aesara.tensor.random.basic

aesara-devs · Dec 15, 2022 · b48a6bb · b48a6bb
1 parent 3418496
commit b48a6bb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/aesara/tensor/random/basic.py b/aesara/tensor/random/basic.py
@@ -2086,6 +2086,9 @@ def __call__(self, x, **kwargs):
 permutation = PermutationRV()
 
 
+random = uniform
+
+
 __all__ = [
     "permutation",
     "choice",
@@ -2126,4 +2129,5 @@ def __call__(self, x, **kwargs):
     "negative_binomial",
     "gengamma",
     "t",
+    "random",
 ]
diff --git a/tests/tensor/random/test_basic.py b/tests/tensor/random/test_basic.py
@@ -47,6 +47,7 @@
     permutation,
     poisson,
     randint,
+    random,
     standard_normal,
     t,
     triangular,
@@ -303,6 +304,10 @@ def test_normal_default_args():
     compare_sample_values(standard_normal)
 
 
+def test_random_default_args():
+    compare_sample_values(random)
+
+
 @pytest.mark.parametrize(
     "mean, sigma, size",
     [