提交 4077c47d authored 作者: Sudarsan Mansingh's avatar Sudarsan Mansingh 提交者: Thomas Wiecki

Fix Argument name

上级 6c4d66c7
...@@ -1615,8 +1615,8 @@ def list_of_nodes( ...@@ -1615,8 +1615,8 @@ def list_of_nodes(
) )
def is_in_ancestors(l_apply: Apply, f_node: Apply) -> bool: def is_in_ancestors(l_apply: Apply, f_apply: Apply) -> bool:
"""Determine if `f_node` is in the graph given by `l_apply`. """Determine if `f_apply` is in the graph given by `l_apply`.
Parameters Parameters
---------- ----------
...@@ -1638,7 +1638,7 @@ def is_in_ancestors(l_apply: Apply, f_node: Apply) -> bool: ...@@ -1638,7 +1638,7 @@ def is_in_ancestors(l_apply: Apply, f_node: Apply) -> bool:
continue continue
if all(i in computed or i.owner is None for i in cur.inputs): if all(i in computed or i.owner is None for i in cur.inputs):
computed.update(cur.outputs) computed.update(cur.outputs)
if cur is f_node: if cur is f_apply:
return True return True
else: else:
todo.append(cur) todo.append(cur)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论