Fix deepcopy of shared variable due to a Numpy problem.

theano/compile/debugmode.py

@@ -847,6 +847,7 @@ def _lessbroken_deepcopy(a):
     called on a 0-d array will return a numpy scalar, not an array.
     """
     # this exists because copy.deepcopy on numpy arrays is broken
+    # This logic is also in link.py
     if type(a) in (numpy.ndarray, numpy.memmap):
         rval = a.copy()
     else:

theano/gof/link.py

 """WRITEME"""
-from copy import copy
+from copy import copy, deepcopy
 import StringIO
 import sys
 import traceback
 
+import numpy
+
 import theano
 from theano.gof import utils
 from theano.gof import graph
@@ -318,6 +320,24 @@ class Container(object):
     def __repr__(self):
         return "<" + repr(self.storage[0]) + ">"
 
+    def __deepcopy__(self, memo):
+        # this exists because copy.deepcopy on numpy arrays is broken
+        a = self.storage[0]
+        if type(a) in (numpy.ndarray, numpy.memmap):
+            a = a.copy()
+        else:
+            a = copy.deepcopy(a)
+
+        r = type(self)(
+            deepcopy(self.type, memo),
+            [a],
+            deepcopy(self.readonly),
+            deepcopy(self.strict),
+            deepcopy(self.allow_downcast),
+            deepcopy(self.name, memo),
+            )
+        return r
+
 
 def map_storage(fgraph, order, input_storage, output_storage):
     """Ensure there is storage (a length-1 list) for inputs, outputs, and interior nodes.

theano/gof/tests/test_link.py

+from copy import deepcopy
 import unittest
 
+import numpy
+
 from theano.gof import graph
 from theano.gof.graph import Variable, Apply, Constant
 from theano.gof.type import Type
@@ -9,6 +12,7 @@ from theano.gof import fg
 from theano.gof.link import *
 from theano.compat import cmp
 
+
 def as_variable(x):
     assert isinstance(x, Variable)
     return x
@@ -178,3 +182,15 @@ def test_sort_schedule_fn():
     for a, b in zip(nodes[:-1], nodes[1:]):
         if not depends((b,a)):
             assert str(a) < str(b)
+
+
+def test_container_deepcopy():
+    """
+    This is a test to a work around a NumPy.
+    """
+    t = theano.tensor.scalar()
+    v = numpy.asarray(0.)
+    c = Container(t, [v])
+    assert isinstance(c.storage[0], numpy.ndarray)
+    deepcopy(c)
+    assert isinstance(c.storage[0], numpy.ndarray)