add random module

(cherry picked from commit 38ff784)
ferrine · Jun 24, 2017 · 378a74b · 378a74b
1 parent eb69131
commit 378a74b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/gelato/__init__.py b/gelato/__init__.py
@@ -1,4 +1,5 @@
-from pymc3.theanof import set_tt_rng, tt_rng
+from . import random
+from .random import *
 from gelato.specs import *
 from . import layers
 from . import specs

diff --git a/gelato/random.py b/gelato/random.py
@@ -0,0 +1,5 @@
+from pymc3.theanof import set_tt_rng, tt_rng
+__all__ = [
+    'set_tt_rng',
+    'tt_rng'
+]