disable info output

This commit is contained in:
QuanyiLi
2023-05-08 13:40:52 +01:00
parent 7ffbf763bd
commit 9483b8fc56

View File

@@ -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]