modified register_specialize and register_canonicalize in tensor.opt to return

the optimization they register. This way they can be used as decorators without accidentally making their optimization disappear from the defining module.

modified register_specialize and register_canonicalize in tensor.opt to return
6a1110bd · James Bergstra · d9174295 · 6a1110bd
--- a/theano/tensor/opt.py
+++ b/theano/tensor/opt.py
@@ -108,10 +108,12 @@ compile.optdb.register('inplace_opt', insert_inplace_optimizer, 75, 'fast_run',
 def register_canonicalize(lopt, *tags, **kwargs):
    name = (kwargs and kwargs.pop('name')) or lopt.__name__
    compile.optdb['canonicalize'].register(name, lopt, 'fast_run', *tags)
+    return lopt
 def register_specialize(lopt, *tags, **kwargs):
    name = (kwargs and kwargs.pop('name')) or lopt.__name__
    compile.optdb['specialize'].register(name, lopt, 'fast_run', *tags)
+    return lopt
 ######################
 # DimShuffle lifters #