diff --git a/scenarionet/converter/pg/utils.py b/scenarionet/converter/pg/utils.py index f59f600..1de52b0 100644 --- a/scenarionet/converter/pg/utils.py +++ b/scenarionet/converter/pg/utils.py @@ -1,4 +1,5 @@ from metadrive.envs.metadrive_env import MetaDriveEnv +import logging from metadrive.scenario.scenario_description import ScenarioDescription as SD @@ -9,6 +10,7 @@ def convert_pg_scenario(scenario_index, version, env): :param version: place holder :param env: metadrive env instance """ + logging.disable(logging.INFO) policy = lambda x: [0, 1] # placeholder scenarios, done_info = env.export_scenarios(policy, scenario_index=[scenario_index], to_dict=False) scenario = scenarios[scenario_index]