提交 b7f4e2c3 authored 作者: Frederic's avatar Frederic

Fix pydotprint when the fgraph.outputs is reused.

上级 514c7de4
...@@ -886,7 +886,7 @@ def pydotprint(fct, outfile=None, ...@@ -886,7 +886,7 @@ def pydotprint(fct, outfile=None,
else: else:
g.add_node(pd.Node(varstr, color='green', shape=var_shape)) g.add_node(pd.Node(varstr, color='green', shape=var_shape))
g.add_edge(pd.Edge(varstr, astr, label=label, **param)) g.add_edge(pd.Edge(varstr, astr, label=label, **param))
elif var.name or not compact: elif var.name or not compact or var in outputs:
g.add_edge(pd.Edge(varstr, astr, label=label, **param)) g.add_edge(pd.Edge(varstr, astr, label=label, **param))
else: else:
# no name, so we don't make a var ellipse # no name, so we don't make a var ellipse
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论