From 378a74b09282d1139bdc2558c646380a34f012a8 Mon Sep 17 00:00:00 2001
From: Maxim Kochurov <maxim.v.kochurov@gmail.com>
Date: Sat, 24 Jun 2017 20:54:12 +0300
Subject: [PATCH] add random module

(cherry picked from commit 38ff784)
---
 gelato/__init__.py | 3 ++-
 gelato/random.py   | 5 +++++
 2 files changed, 7 insertions(+), 1 deletion(-)
 create mode 100644 gelato/random.py

diff --git a/gelato/__init__.py b/gelato/__init__.py
index afb2596..25d1fd6 100644
--- 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
new file mode 100644
index 0000000..a14c2ac
--- /dev/null
+++ b/gelato/random.py
@@ -0,0 +1,5 @@
+from pymc3.theanof import set_tt_rng, tt_rng
+__all__ = [
+    'set_tt_rng',
+    'tt_rng'
+]