98f7bac81b00256bc31e66aab2d7752bc320e67c,odps/df/backends/optimize/utils.py,,copy_sequence,#Any#Any#Any#,44

Before Change

            curr.substitute(seq, copied_seq, dag=dag)
            curr = copied_seq

    return copied

After Change

def copy_sequence(sequence, collection, dag=None):
    if dag is None:
        dag = sequence.to_dag(copy=False)

    traversed = set()
    copies = dict()

    def copy(node):
        if node._node_id not in copies:
            copies[node._node_id] = node.copy()
        copied = copies[node._node_id]
        if not dag.contains_node(copied):
            dag.add_node(copied)
        return copied

    for n in sequence.traverse(top_down=True, traversed=traversed,
                               stop_cond=lambda x: x is collection, unique=True):
        if n is sequence:
            copy(n)
            continue
        if n is collection:
            continue

        parents = [p for p in dag.successors(n) if p._node_id in traversed]
        for parent in parents:
            copy(parent).substitute(n, copy(n), dag=dag)

    return copies[sequence._node_id]

Instances