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

Fix pydotprint when the fgraph.outputs is reused.

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