From b5418ba3f6a02c475dd5ac726a3b6518f1dc794a Mon Sep 17 00:00:00 2001 From: QuanyiLi Date: Mon, 8 May 2023 17:07:24 +0100 Subject: [PATCH] fix bug --- scenarionet/scripts/verify_dataset.py | 6 ++++-- scenarionet/tests/local_test/combine_verify_generate.sh | 6 +++--- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/scenarionet/scripts/verify_dataset.py b/scenarionet/scripts/verify_dataset.py index 9797190..89f3149 100644 --- a/scenarionet/scripts/verify_dataset.py +++ b/scenarionet/scripts/verify_dataset.py @@ -4,12 +4,14 @@ MetaDrive. """ import argparse -from scenarionet.verifier.utils import verify_loading_into_metadrive +from scenarionet.verifier.utils import verify_loading_into_metadrive, set_random_drop 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") parser.add_argument("--num_workers", type=int, default=8, help="number of workers to use") + parser.add_argument("--random_drop", action="store_true", help="Randomly make some scenarios fail. for test only!") args = parser.parse_args() - verify_loading_into_metadrive(args.dataset_path, args.result_save_dir, num_workers=args.num_workers) + set_random_drop(args.random_drop) + verify_loading_into_metadrive(args.dataset_path, args.result_save_dir, num_workers=args.num_workers) diff --git a/scenarionet/tests/local_test/combine_verify_generate.sh b/scenarionet/tests/local_test/combine_verify_generate.sh index bf6c9d7..272b920 100644 --- a/scenarionet/tests/local_test/combine_verify_generate.sh +++ b/scenarionet/tests/local_test/combine_verify_generate.sh @@ -1,6 +1,6 @@ #!/usr/bin/env bash python ../../scripts/combine_dataset.py --to ../tmp/test_combine_dataset --from_datasets ../../../dataset/waymo ../../../dataset/pg ../../../dataset/nuscenes ../../../dataset/nuplan --overwrite -python ../../scripts/verify_dataset.py --dataset_path ../tmp/test_combine_dataset --result_save_dir ../tmp/test_combine_dataset -python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/errors_test_combine_dataset.json --overwrite --dataset_path ../tmp/verify_pass -python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/errors_test_combine_dataset.json --overwrite --dataset_path ../tmp/verify_fail --broken \ No newline at end of file +python ../../scripts/verify_dataset.py --dataset_path ../tmp/test_combine_dataset --result_save_dir ../tmp/test_combine_dataset --random_drop +python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/error_scenarios_for_test_combine_dataset.json --overwrite --dataset_path ../tmp/verify_pass +python ../../scripts/generate_from_error_file.py --file ../tmp/test_combine_dataset/errors_scenarios_for_combine_dataset.json --overwrite --dataset_path ../tmp/verify_fail --broken \ No newline at end of file