提交 fdeabc18 authored 作者: Virgile Andreani's avatar Virgile Andreani 提交者: Ricardo Vieira

Make non-strict zip strict in printing.py

上级 70e4e579
...@@ -930,7 +930,7 @@ class PatternPrinter(Printer): ...@@ -930,7 +930,7 @@ class PatternPrinter(Printer):
) )
idx = node.outputs.index(output) idx = node.outputs.index(output)
pattern, precedences = self.patterns[idx] pattern, precedences = self.patterns[idx]
precedences += (1000,) * len(node.inputs) precedences += (1000,) * (len(node.inputs) - len(precedences))
def pp_process(input, new_precedence): def pp_process(input, new_precedence):
with set_precedence(pstate, new_precedence): with set_precedence(pstate, new_precedence):
...@@ -938,10 +938,9 @@ class PatternPrinter(Printer): ...@@ -938,10 +938,9 @@ class PatternPrinter(Printer):
return r return r
d = { d = {
str(i): x str(i): pp_process(input, precedence)
for i, x in enumerate( for i, (input, precedence) in enumerate(
pp_process(input, precedence) zip(node.inputs, precedences, strict=True)
for input, precedence in zip(node.inputs, precedences, strict=False)
) )
} }
r = pattern % d r = pattern % d
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论