From e37d15c739d35361a564e0705e39cb1110483453 Mon Sep 17 00:00:00 2001 From: Min Shi Date: Wed, 27 Mar 2024 10:44:21 +0800 Subject: [PATCH] use apped instead of remove --- src/promptflow/promptflow/batch/_batch_engine.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/promptflow/promptflow/batch/_batch_engine.py b/src/promptflow/promptflow/batch/_batch_engine.py index a11a9d2dc107..5c9b9183cfc4 100644 --- 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