diff --git a/scenarionet/scripts/verify_scenarios.py b/scenarionet/scripts/verify_scenarios.py index e69de29..3c3a8aa 100644 --- a/scenarionet/scripts/verify_scenarios.py +++ b/scenarionet/scripts/verify_scenarios.py @@ -0,0 +1,14 @@ +""" +This script aims to convert nuscenes scenarios to ScenarioDescription, so that we can load any nuscenes scenarios into +MetaDrive. +""" +import argparse + +from scenarionet.verifier.utils import verify_loading_into_metadrive + +if __name__ == '__main__': + parser = argparse.ArgumentParser() + parser.add_argument("--dataset_path", required=True, help="Dataset path, a directory") + parser.add_argument("--result_save_dir", required=True, help="Dataset path, a directory") + args = parser.parse_args() + verify_loading_into_metadrive(args.dataset_path, args.result_save_dir) diff --git a/scenarionet/tests/local_test/_test_combine_dataset_local.py b/scenarionet/tests/local_test/_test_combine_dataset_local.py index 7045d14..d8eee56 100644 --- a/scenarionet/tests/local_test/_test_combine_dataset_local.py +++ b/scenarionet/tests/local_test/_test_combine_dataset_local.py @@ -16,8 +16,8 @@ def _test_combine_dataset(): combine_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_dataset") combine_multiple_dataset(combine_path, *dataset_paths, force_overwrite=True, try_generate_missing_file=True) os.makedirs("verify_results", exist_ok=True) - success, result = verify_loading_into_metadrive(combine_path, "verify_results") - assert success + verify_loading_into_metadrive(combine_path, "verify_results") + # assert success if __name__ == '__main__': diff --git a/scenarionet/tests/local_test/combine_and_verify.sh b/scenarionet/tests/local_test/combine_and_verify.sh new file mode 100644 index 0000000..7fad21e --- /dev/null +++ b/scenarionet/tests/local_test/combine_and_verify.sh @@ -0,0 +1,4 @@ +#!/usr/bin/env bash + +python ../../scripts/combine_dataset.py --to_dataset ../../dataset/test_combine_dataset --from_dataset ../../dataset/waymo ../../dataset/pg ../../dataset/nuscenes ../../dataset/nuplan --overwrite +python ../../scripts/combine_dataset.py --dataset_path ../../dataset/test_combine_dataset --result_save_dir ../../dataset/test_combine_dataset \ No newline at end of file diff --git a/scenarionet/tests/local_test/combine_data.sh b/scenarionet/tests/local_test/combine_data.sh deleted file mode 100644 index 17ca206..0000000 --- a/scenarionet/tests/local_test/combine_data.sh +++ /dev/null @@ -1,3 +0,0 @@ -#!/usr/bin/env bash - -nohup python ../../scripts/combine_dataset.py --to_dataset ../../dataset/test_combine_dataset --from_dataset ../../dataset/waymo ../../dataset/pg ../../dataset/nuscenes ../../dataset/nuplan --overwrite > nuplan.log 2>&1 & \ No newline at end of file