From 300a29da098b132c6d20e88a09ebf8e462dab24e Mon Sep 17 00:00:00 2001 From: QuanyiLi Date: Sat, 6 May 2023 22:43:12 +0100 Subject: [PATCH] fix nuplan bug --- scenarionet/converter/nuplan/utils.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/scenarionet/converter/nuplan/utils.py b/scenarionet/converter/nuplan/utils.py index 4e07744..9462fa2 100644 --- a/scenarionet/converter/nuplan/utils.py +++ b/scenarionet/converter/nuplan/utils.py @@ -50,11 +50,12 @@ def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None] = None, bu :return: """ nuplan_package_path = NUPLAN_PACKAGE_PATH - logs = logs or [file for file in os.listdir(data_root)] log_string = "" for log in logs: - log_string += log[:-3] # remove .db + if log[-3:] == ".db": + log = log[:-3] + log_string += log log_string += "," log_string = log_string[:-1] @@ -356,9 +357,9 @@ def extract_traffic(scenario: NuPlanScenario, center): type=MetaDriveType.UNSET, state=dict( position=np.zeros(shape=(episode_len, 3)), - heading=np.zeros(shape=(episode_len, )), + heading=np.zeros(shape=(episode_len,)), velocity=np.zeros(shape=(episode_len, 2)), - valid=np.zeros(shape=(episode_len, )), + valid=np.zeros(shape=(episode_len,)), length=np.zeros(shape=(episode_len, 1)), width=np.zeros(shape=(episode_len, 1)), height=np.zeros(shape=(episode_len, 1))