From 26013c6fd6de06d16269b621ec0d32cf726ee472 Mon Sep 17 00:00:00 2001 From: QuanyiLi Date: Sun, 7 May 2023 17:31:00 +0100 Subject: [PATCH] remove logging information --- scenarionet/tests/_test_combine_dataset_local.py | 2 +- scenarionet/verifier/utils.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/scenarionet/tests/_test_combine_dataset_local.py b/scenarionet/tests/_test_combine_dataset_local.py index f41e79e..a4243ea 100644 --- a/scenarionet/tests/_test_combine_dataset_local.py +++ b/scenarionet/tests/_test_combine_dataset_local.py @@ -13,7 +13,7 @@ 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) - success, result = verify_loading_into_metadrive(combine_path) + success, result = verify_loading_into_metadrive(combine_path, steps_to_run=250) assert success diff --git a/scenarionet/verifier/utils.py b/scenarionet/verifier/utils.py index 65674fe..fe0cef9 100644 --- a/scenarionet/verifier/utils.py +++ b/scenarionet/verifier/utils.py @@ -10,7 +10,7 @@ from metadrive.scenario.utils import get_number_of_scenarios def verify_loading_into_metadrive(dataset_path, result_save_dir=None, steps_to_run=0): - print("================ Begin Scenario Loading oVerification ================ \n") + print("================ Begin Scenario Loading Verification for {} ================ \n".format(result_save_dir)) scenario_num = get_number_of_scenarios(dataset_path) if result_save_dir is not None: assert os.path.exists(result_save_dir) and os.path.isdir( @@ -25,6 +25,7 @@ def verify_loading_into_metadrive(dataset_path, result_save_dir=None, steps_to_r "data_directory": dataset_path, } ) + logging.disable(logging.WARNING) error_files = [] try: for i in tqdm.tqdm(range(scenario_num)):