提交 09762a8b authored 作者: Frederic Bastien's avatar Frederic Bastien

Make intermediate var as a box as input/output/shared var

上级 7830b424
......@@ -957,6 +957,10 @@ def pydotprint(fct, outfile=None,
label = label[:max_label_size - 3] + '...'
param['label'] = label
g.add_edge(pd.Edge(astr, varstr, **param))
if high_contrast:
g.add_node(pd.Node(varstr, shape=var_shape))
else:
g.add_node(pd.Node(varstr, shape=var_shape))
# else:
# don't add egde here as it is already added from the inputs.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论