diff --git a/scenarionet/converter/nuplan/utils.py b/scenarionet/converter/nuplan/utils.py index 7defe86..664c3da 100644 --- a/scenarionet/converter/nuplan/utils.py +++ b/scenarionet/converter/nuplan/utils.py @@ -40,7 +40,7 @@ except ImportError as e: EGO = "ego" -def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None], builder="nuplan_mini"): +def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None] = None, builder="nuplan_mini"): """ Getting scenarios. You could use your parameters to get a bunch of scenarios :param data_root: path contains .db files, like /nuplan-v1.1/splits/mini @@ -54,7 +54,7 @@ def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None], builder=" logs = logs or [file for file in os.listdir(data_root)] log_string = "" for log in logs: - log_string += log + log_string += log[:-3] # remove .db log_string += "," log_string = log_string[:-1] diff --git a/scenarionet/examples/convert_nuplan.py b/scenarionet/examples/convert_nuplan.py index ec52c26..607b4fe 100644 --- a/scenarionet/examples/convert_nuplan.py +++ b/scenarionet/examples/convert_nuplan.py @@ -16,7 +16,8 @@ if __name__ == "__main__": data_root = os.path.join(os.getenv("NUPLAN_DATA_ROOT"), "nuplan-v1.1/splits/mini") map_root = os.getenv("NUPLAN_MAPS_ROOT") - scenarios = get_nuplan_scenarios(data_root, map_root, logs=["2021.07.16.20.45.29_veh-35_01095_01486"]) + # scenarios = get_nuplan_scenarios(data_root, map_root, logs=["2021.07.16.20.45.29_veh-35_01095_01486"]) + scenarios = get_nuplan_scenarios(data_root, map_root) write_to_directory(convert_func=convert_nuplan_scenario, scenarios=scenarios,