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

disable info output

Browse Source
This commit is contained in:
QuanyiLi
2023-05-08 13:40:52 +01:00
parent 7ffbf763bd
commit 9483b8fc56
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scenarionet/converter/pg/utils.py
View File

@@ -1,4 +1,5 @@
from metadrive.envs.metadrive_env import MetaDriveEnv from metadrive.envs.metadrive_env import MetaDriveEnv
import logging
from metadrive.scenario.scenario_description import ScenarioDescription as SD 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 version: place holder
:param env: metadrive env instance :param env: metadrive env instance
""" """
logging.disable(logging.INFO)
policy = lambda x: [0, 1] # placeholder policy = lambda x: [0, 1] # placeholder
scenarios, done_info = env.export_scenarios(policy, scenario_index=[scenario_index], to_dict=False) scenarios, done_info = env.export_scenarios(policy, scenario_index=[scenario_index], to_dict=False)
scenario = scenarios[scenario_index] scenario = scenarios[scenario_index]