Created wrapper code

theano/compile/dictionaryOutputWrapper.py

 '''
-Supports user generating theano function where the outputs are a dictionary
+Generates a wrapper around theano functions that allows the user to receive outputs in a dictionary.  
 '''
 
-def createFunctionReturningDictionary(args, fn, keys):
-    outputLst = fn(*args)
+def createFunctionReturningDictionary(args, kwargs, fn, keys):
+
+    outputLst = fn(*args, **kwargs)
 
     outputDict = {}
 
@@ -12,3 +13,4 @@ def createFunctionReturningDictionary(args, fn, keys):
 
     return outputDict
 
+

theano/compile/function.py

@@ -280,7 +280,7 @@ def function(inputs, outputs=None, mode=None, updates=None, givens=None,
 
     if outputsDictFormat:
 
-        fnDictOutput = (lambda *args: createFunctionReturningDictionary(args, fn = fn, keys = outputKeys))
+        fnDictOutput = (lambda *args, **kwargs: createFunctionReturningDictionary(args, kwargs, fn = fn, keys = outputKeys))
 
         return fnDictOutput