huangfu/scenarionet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove logging information

Browse Source
This commit is contained in:
QuanyiLi
2023-05-07 17:31:00 +01:00
parent b0791e5136
commit 26013c6fd6
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scenarionet/tests/_test_combine_dataset_local.py
View File

@@ -13,7 +13,7 @@ def _test_combine_dataset():
combine_path = os.path.join(SCENARIONET_DATASET_PATH, "combined_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) 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 assert success

3
scenarionet/verifier/utils.py
View File

@@ -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): 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) scenario_num = get_number_of_scenarios(dataset_path)
if result_save_dir is not None: if result_save_dir is not None:
assert os.path.exists(result_save_dir) and os.path.isdir( 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, "data_directory": dataset_path,
} }
) )
logging.disable(logging.WARNING)
error_files = [] error_files = []
try: try:
for i in tqdm.tqdm(range(scenario_num)): for i in tqdm.tqdm(range(scenario_num)):