Mercurial > pylearn

changeset 906:d1a757d17e19

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
fix some import.
author Frederic Bastien <nouiz@nouiz.org>
date Thu, 18 Mar 2010 10:49:09 -0400
parents 70b061f142b2
children 6d4f98f86514
files pylearn/sandbox/test_scan_inputs_groups.py
diffstat 1 files changed, 4 insertions(+), 3 deletions(-) [+]
line wrap: on
line diff
--- a/pylearn/sandbox/test_scan_inputs_groups.py	Thu Mar 18 10:45:29 2010 -0400
+++ b/pylearn/sandbox/test_scan_inputs_groups.py	Thu Mar 18 10:49:09 2010 -0400
@@ -9,6 +9,7 @@
 import theano.tensor as T
 from pylearn.sandbox.scan_inputs_groups import FillMissing
 import theano.compile.mode as mode_module
+import theano
 
 class TestFillMissing(unittest.TestCase):
     def setUp(self):
@@ -16,9 +17,9 @@
 
         #we need to desactivate the check for NaN value as we have them in input
         #TODO: Make an option to don't check NaN value in input only, bug check in output.
-        m=mode_module.default_mode
-        if m=="DEBUG_MODE":
-            m=copy.copy(mode_module.predefined_modes[m])
+        m=mode_module.get_default_mode()
+        if isinstance(m,theano.compile.debugmode.DebugMode):
+            m=copy.copy(m)
             m.check_isfinite=False
         self.mode = m