Remove weak references, which aren't picklable

theano/sandbox/rng_mrg.py

@@ -6,7 +6,6 @@ http://www.iro.umontreal.ca/~simardr/ssj/indexe.html
 
 """
 import warnings
-import weakref
 
 import numpy
 
@@ -1222,7 +1221,7 @@ class MRG_RandomStreams(object):
                                    *mrg_uniform.new(node_rstate,
                                                     ndim, dtype, size))
         # Add a reference to distinguish from other shared variables
-        node_rstate.rng_owner = weakref.ref(self)
+        node_rstate.tag.is_rng = True
         r = u * (high - low) + low
 
         if u.type.broadcastable != r.type.broadcastable:

theano/tensor/shared_randomstreams.py

@@ -5,7 +5,6 @@ __docformat__ = "restructuredtext en"
 
 import copy
 import numpy
-import weakref
 
 from theano.compile.sharedvalue import (SharedVariable, shared_constructor,
                                         shared)
@@ -134,7 +133,7 @@ class RandomStreams(raw_random.RandomStreamsBase):
         seed = int(self.gen_seedgen.randint(2 ** 30))
         random_state_variable = shared(numpy.random.RandomState(seed))
         # Add a reference to distinguish from other shared variables
-        random_state_variable.rng_owner = weakref.ref(self)
+        random_state_variable.tag.is_rng = True
         new_r, out = op(random_state_variable, *args, **kwargs)
         out.rng = random_state_variable
         out.update = (random_state_variable, new_r)