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)):