From 52ff95a07822f26fe45ff106538345815e390437 Mon Sep 17 00:00:00 2001 From: QuanyiLi Date: Tue, 8 Aug 2023 17:51:04 +0100 Subject: [PATCH] format --- scenarionet/tests/script/run_ckpt.py | 70 ++++++++++++++-------------- 1 file changed, 35 insertions(+), 35 deletions(-) diff --git a/scenarionet/tests/script/run_ckpt.py b/scenarionet/tests/script/run_ckpt.py index 0086ce2..ec5e134 100644 --- a/scenarionet/tests/script/run_ckpt.py +++ b/scenarionet/tests/script/run_ckpt.py @@ -7,41 +7,41 @@ from scenarionet_training.scripts.train_nuplan import config if __name__ == "__main__": initialize_ray(test_mode=False, num_gpus=1) - env = createGymWrapper(ScenarioEnv)({ - # "data_directory": AssetLoader.file_path("nuscenes", return_raw_style=False), - "data_directory": "D:\\scenarionet_testset\\nuplan_test\\nuplan_test_w_raw", - "use_render": True, - # "agent_policy": ReplayEgoCarPolicy, - "show_interface": False, - "image_observation": False, - "show_logo": False, - "no_traffic": False, - "no_static_vehicles": False, - "drivable_region_extension": 15, - "sequential_seed": True, - "reactive_traffic": True, - "show_fps": False, - "render_pipeline": True, - "daytime": "07:10", - "max_lateral_dist": 2, - "window_size": (1200, 800), - "camera_dist": 9, - "start_scenario_index": 5, - "num_scenarios": 4000, - "horizon": 1000, - "store_map": False, - "vehicle_config": dict( - show_navi_mark=True, - # no_wheel_friction=True, - use_special_color=False, - image_source="depth_camera", - lidar=dict(num_lasers=120, distance=50), - lane_line_detector=dict(num_lasers=0, distance=50), - side_detector=dict(num_lasers=0, distance=50) - ), - } - ) - + env = createGymWrapper(ScenarioEnv)( + { + # "data_directory": AssetLoader.file_path("nuscenes", return_raw_style=False), + "data_directory": "D:\\scenarionet_testset\\nuplan_test\\nuplan_test_w_raw", + "use_render": True, + # "agent_policy": ReplayEgoCarPolicy, + "show_interface": False, + "image_observation": False, + "show_logo": False, + "no_traffic": False, + "no_static_vehicles": False, + "drivable_region_extension": 15, + "sequential_seed": True, + "reactive_traffic": True, + "show_fps": False, + "render_pipeline": True, + "daytime": "07:10", + "max_lateral_dist": 2, + "window_size": (1200, 800), + "camera_dist": 9, + "start_scenario_index": 5, + "num_scenarios": 4000, + "horizon": 1000, + "store_map": False, + "vehicle_config": dict( + show_navi_mark=True, + # no_wheel_friction=True, + use_special_color=False, + image_source="depth_camera", + lidar=dict(num_lasers=120, distance=50), + lane_line_detector=dict(num_lasers=0, distance=50), + side_detector=dict(num_lasers=0, distance=50) + ), + } + ) # env.reset() #