提交 b8772f4f authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix a test with python 3

上级 7ffd86a0
......@@ -55,7 +55,7 @@ def test_pydotprint_return_image():
x = tensor.dvector()
ret = theano.printing.pydotprint(x * 2, return_image=True)
assert isinstance(ret, str)
assert isinstance(ret, (str, bytes))
def test_pydotprint_variables():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论