use apped instead of remove

microsoft · Mar 27, 2024 · e37d15c · e37d15c
1 parent 220ffdd
commit e37d15c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/promptflow/promptflow/batch/_batch_engine.py b/src/promptflow/promptflow/batch/_batch_engine.py
@@ -252,9 +252,12 @@ def _copy_previous_run_result(
 
                     # Load previous node run info and remove aggregation nodes in case it is loaded into node run info
                     previous_node_run_infos = resume_from_run_storage.load_node_run_info_for_line(i)
-                    for node_run_info in previous_node_run_infos:
-                        if node_run_info.node in aggregation_nodes:
-                            previous_node_run_infos.remove(node_run_info)
+                    if aggregation_nodes:
+                        previous_node_run_infos = [
+                            node_run_info
+                            for node_run_info in previous_node_run_infos
+                            if node_run_info.node not in aggregation_nodes
+                        ]
                     previous_node_run_infos_dict = {node_run.node: node_run for node_run in previous_node_run_infos}
                     previous_node_run_outputs = {
                         node_info.node: node_info.output for node_info in previous_node_run_infos