Fix hierarchy for pipeline nodes without args

This commit is contained in:
allegroai 2024-07-29 17:37:05 +03:00
parent 2b66ee663e
commit 1b474dc0b0

View File

@ -4805,7 +4805,7 @@ class PipelineDecorator(PipelineController):
if n not in _node.parents:
_node.parents.append(n)
break
if kwargs:
leaves = cls._singleton._find_executed_node_leaves()
_node.parents = (_node.parents or []) + [
x for x in cls._evaluated_return_values.get(tid, []) if x in leaves