From 12132bc870e166af05d314ceed7d0499697b04d0 Mon Sep 17 00:00:00 2001 From: QuanyiLi Date: Sun, 21 May 2023 14:45:01 +0100 Subject: [PATCH] remove repeat calculation --- scenarionet/converter/utils.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/scenarionet/converter/utils.py b/scenarionet/converter/utils.py index 106455b..704c4cc 100644 --- a/scenarionet/converter/utils.py +++ b/scenarionet/converter/utils.py @@ -211,12 +211,8 @@ def write_to_directory_single_worker( # add agents summary summary_dict = {} - ego_car_id = sd_scenario[SD.METADATA][SD.SDC_ID] - summary_dict[ego_car_id] = SD.get_object_summary( - state_dict=sd_scenario.get_sdc_track()["state"], id=ego_car_id, type=sd_scenario.get_sdc_track()["type"] - ) for track_id, track in sd_scenario[SD.TRACKS].items(): - summary_dict[track_id] = SD.get_object_summary(state_dict=track["state"], id=track_id, type=track["type"]) + summary_dict[track_id] = SD.get_object_summary(state_dict=track, id=track_id) sd_scenario[SD.METADATA][SD.SUMMARY.OBJECT_SUMMARY] = summary_dict # count some objects occurrence