cosmatic changes

theano/compile/function_module.py

@@ -1221,15 +1221,12 @@ class FunctionMaker(object):
 
         if fgraph is None:
             need_opt = True
-        else:
-            need_opt = False
-            
-        if fgraph is None:
             # make the fgraph (copies the graph, creates NEW INPUT AND OUTPUT VARIABLES)
             fgraph, additional_outputs = std_fgraph(inputs, outputs, accept_inplace)
             fgraph.profile = profile
         else:
             # fgraph is already an optimized one
+            need_opt = False
             _, additional_outputs = std_fgraph(inputs, outputs, accept_inplace)
             pass
         
@@ -1239,11 +1236,10 @@ class FunctionMaker(object):
         optimizer, linker = mode.optimizer, copy.copy(mode.linker)
 
         compute_test_value_orig = theano.config.compute_test_value
-        add_stack_trace_on_call = gof.Op.add_stack_trace_on_call
+        add_stack_trace_on_call_orig = gof.Op.add_stack_trace_on_call
         
         if need_opt:
             # optimize the fgraph
-            try:
             theano.config.compute_test_value = theano.config.compute_test_value_opt
             gof.Op.add_stack_trace_on_call = False
             start_optimizer = time.time()
@@ -1258,14 +1254,9 @@ class FunctionMaker(object):
 
             #Add deep copy to respect the memory interface
             insert_deepcopy(fgraph, inputs, outputs + additional_outputs)
-            finally:
-                theano.config.compute_test_value = compute_test_value_orig
-                gof.Op.add_stack_trace_on_call = add_stack_trace_on_call
-                
-        else:
         # fgraph is already optimized
         theano.config.compute_test_value = compute_test_value_orig
-            gof.Op.add_stack_trace_on_call = add_stack_trace_on_call
+        gof.Op.add_stack_trace_on_call = add_stack_trace_on_call_orig
         
         # initialize the linker
         if not hasattr(linker, 'accept'):

theano/tests/test_pickle_unpickle_theano_fn.py

@@ -15,7 +15,7 @@ This note is written by Li Yao.
 import unittest
 import numpy
 import cPickle
-from collections import OrderedDict
+from theano.compat.python2x import DictMixin, OrderedDict
 floatX = 'float32'
 import theano
 import theano.tensor as T