提交 62b2dc0e authored 作者: Frederic Bastien's avatar Frederic Bastien

Another theano.pp fix as gh-5389

上级 074d7c2d
...@@ -834,9 +834,14 @@ class MakeVectorPrinter: ...@@ -834,9 +834,14 @@ class MakeVectorPrinter:
if r.owner is None: if r.owner is None:
raise TypeError("Can only print make_vector.") raise TypeError("Can only print make_vector.")
elif isinstance(r.owner.op, MakeVector): elif isinstance(r.owner.op, MakeVector):
return "[%s]" % ", ".join( old_precedence = getattr(pstate, 'precedence', None)
pstate.pprinter.process(input, pstate.clone(precedence=1000)) try:
for input in r.owner.inputs) pstate.precedence = 1000
s = [pstate.pprinter.process(input)
for input in r.owner.inputs]
finally:
pstate.precedence = old_precedence
return "[%s]" % ", ".join(s)
else: else:
raise TypeError("Can only print make_vector.") raise TypeError("Can only print make_vector.")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论