Adds dirichlet to __init__.py, making it usable

thisiscam · Oct 11, 2019 · 245f406 · 245f406
1 parent 27afa12
commit 245f406
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jax/scipy/stats/__init__.py b/jax/scipy/stats/__init__.py
@@ -16,6 +16,7 @@
 from . import bernoulli
 from . import beta
 from . import cauchy
+from . import dirichlet
 from . import expon
 from . import gamma
 from . import laplace