From bbef2db107df5cc32a38cea0d9efa2fb73ae76d2 Mon Sep 17 00:00:00 2001 From: QuanyiLi Date: Mon, 8 May 2023 15:17:13 +0100 Subject: [PATCH] set waymo origin path --- .gitignore | 1 + scenarionet/scripts/convert_waymo.py | 2 +- scenarionet/tests/local_test/_test_convert_waymo.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 759a188..667439c 100644 --- a/.gitignore +++ b/.gitignore @@ -16,3 +16,4 @@ dataset/* **/tmp/** **/failed_scenarios/ **/passed_senarios/ +**/waymo_origin diff --git a/scenarionet/scripts/convert_waymo.py b/scenarionet/scripts/convert_waymo.py index d1c0d73..43ef132 100644 --- a/scenarionet/scripts/convert_waymo.py +++ b/scenarionet/scripts/convert_waymo.py @@ -23,7 +23,7 @@ if __name__ == '__main__': output_path = args.dataset_path version = args.version - waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "waymo_origin") + waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "../waymo_origin") scenarios = get_waymo_scenarios(waymo_data_directory) write_to_directory( diff --git a/scenarionet/tests/local_test/_test_convert_waymo.py b/scenarionet/tests/local_test/_test_convert_waymo.py index 043e1bc..cc80fca 100644 --- a/scenarionet/tests/local_test/_test_convert_waymo.py +++ b/scenarionet/tests/local_test/_test_convert_waymo.py @@ -22,7 +22,7 @@ if __name__ == '__main__': output_path = args.dataset_path version = args.version - waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "waymo_origin") + waymo_data_directory = os.path.join(SCENARIONET_DATASET_PATH, "../waymo_origin") scenarios = get_waymo_scenarios(waymo_data_directory, num_workers=3) write_to_directory(