提交 a8d0d9c4 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Remove another warning message from the tests' output

上级 4ae9cf4b
...@@ -735,6 +735,8 @@ def test_pickle_aliased_memory(): ...@@ -735,6 +735,8 @@ def test_pickle_aliased_memory():
sio = StringIO.StringIO() sio = StringIO.StringIO()
handler = logging.StreamHandler(sio) handler = logging.StreamHandler(sio)
logging.getLogger('theano.compile.function_module').addHandler(handler) logging.getLogger('theano.compile.function_module').addHandler(handler)
# Silence original handler when intentionnally generating warning messages
logging.getLogger('theano').removeHandler(theano.logging_default_handler)
try: try:
m.f.pickle_aliased_memory_strategy = 'warn' m.f.pickle_aliased_memory_strategy = 'warn'
m.g.pickle_aliased_memory_strategy = 'warn' m.g.pickle_aliased_memory_strategy = 'warn'
...@@ -742,6 +744,7 @@ def test_pickle_aliased_memory(): ...@@ -742,6 +744,7 @@ def test_pickle_aliased_memory():
assert sio.getvalue().startswith('aliased relat') assert sio.getvalue().startswith('aliased relat')
finally: finally:
logging.getLogger('theano.compile.function_module').removeHandler(handler) logging.getLogger('theano.compile.function_module').removeHandler(handler)
logging.getLogger('theano').addHandler(theano.logging_default_handler)
try: try:
m.f.pickle_aliased_memory_strategy = 'raise' m.f.pickle_aliased_memory_strategy = 'raise'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论