提交 6d034f05 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Flake8 for compile/monitormode.py

上级 a19013dd
from __future__ import print_function from __future__ import print_function
# Note: this code was initially copied from the 'pyutools' package by its # Note: this code was initially copied from the 'pyutools' package by its
# original author, and re-licensed under Theano's license. # original author, and re-licensed under Theano's license.
import numpy
import theano import theano
from theano.compile.mode import Mode from theano.compile.mode import Mode
...@@ -48,7 +48,7 @@ class MonitorMode(Mode): ...@@ -48,7 +48,7 @@ class MonitorMode(Mode):
if optimizer == 'default': if optimizer == 'default':
optimizer = theano.config.optimizer optimizer = theano.config.optimizer
if (linker is not None and if (linker is not None and
not isinstance(linker.mode, MonitorMode)): not isinstance(linker.mode, MonitorMode)):
raise Exception("MonitorMode can only use its own linker! You " raise Exception("MonitorMode can only use its own linker! You "
"should not provide one.", linker) "should not provide one.", linker)
...@@ -86,7 +86,7 @@ class MonitorMode(Mode): ...@@ -86,7 +86,7 @@ class MonitorMode(Mode):
def detect_nan(i, node, fn): def detect_nan(i, node, fn):
for output in fn.outputs: for output in fn.outputs:
if (not isinstance(numpy.random.RandomState, output[0]) and if (not isinstance(numpy.random.RandomState, output[0]) and
numpy.isnan(output[0]).any()): numpy.isnan(output[0]).any()):
print('*** NaN detected ***') print('*** NaN detected ***')
theano.printing.debugprint(node) theano.printing.debugprint(node)
print('Inputs : %s' % [input[0] for input in fn.inputs]) print('Inputs : %s' % [input[0] for input in fn.inputs])
......
...@@ -40,7 +40,6 @@ whitelist_flake8 = [ ...@@ -40,7 +40,6 @@ whitelist_flake8 = [
"tests/unittest_tools.py", "tests/unittest_tools.py",
"compile/__init__.py", "compile/__init__.py",
"compile/profiling.py", "compile/profiling.py",
"compile/monitormode.py",
"compile/tests/test_builders.py", "compile/tests/test_builders.py",
"compile/tests/test_misc.py", "compile/tests/test_misc.py",
"compile/tests/test_monitormode.py", "compile/tests/test_monitormode.py",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论