pickling bug

theano/compile/function_module.py

@@ -551,7 +551,7 @@ class FunctionMaker(object):
 
 
 def _pickle_FunctionMaker(fm):
-    return (_constructor_FunctionMaker, (fm.inputs, fm.outputs, fm.mode, fm.accept_inplace))
+    return (_constructor_FunctionMaker, (fm.inputs, fm.outputs[0] if fm.unpack_single else fm.outputs, fm.mode, fm.accept_inplace))
 
 def _constructor_FunctionMaker(*args):
     return FunctionMaker(*args)

theano/compile/module.py

@@ -301,7 +301,7 @@ class CompositeInstance(object):
             x.initialize(value)
         else:
             ##self.__items__[item] = value
-            raise TypeError('Cannot set item %s' % item)
+            raise KeyError('Cannot set item %s' % item)
 
 class Composite(Component):