Tests should not fail on optional dependency.

上级 e2fbb03e
...@@ -18,11 +18,16 @@ pydot_imported = False ...@@ -18,11 +18,16 @@ pydot_imported = False
try: try:
# pydot-ng is a fork of pydot that is better maintained # pydot-ng is a fork of pydot that is better maintained
import pydot_ng as pd import pydot_ng as pd
if pd.find_graphviz():
pydot_imported = True
except ImportError: except ImportError:
try:
# fall back on pydot if necessary # fall back on pydot if necessary
import pydot as pd import pydot as pd
if pd.find_graphviz(): if pd.find_graphviz():
pydot_imported = True pydot_imported = True
except ImportError:
pass # tests should not fail on optional dependency
class PyDotFormatter(object): class PyDotFormatter(object):
......
...@@ -26,11 +26,16 @@ pydot_imported = False ...@@ -26,11 +26,16 @@ pydot_imported = False
try: try:
# pydot-ng is a fork of pydot that is better maintained # pydot-ng is a fork of pydot that is better maintained
import pydot_ng as pd import pydot_ng as pd
if pd.find_graphviz():
pydot_imported = True
except ImportError: except ImportError:
try:
# fall back on pydot if necessary # fall back on pydot if necessary
import pydot as pd import pydot as pd
if pd.find_graphviz(): if pd.find_graphviz():
pydot_imported = True pydot_imported = True
except ImportError:
pass # tests should not fail on optional dependency
_logger = logging.getLogger("theano.printing") _logger = logging.getLogger("theano.printing")
VALID_ASSOC = set(['left', 'right', 'either']) VALID_ASSOC = set(['left', 'right', 'either'])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论