From 880fcd034e971e9bc4f15f1107ee74ce73c3776e Mon Sep 17 00:00:00 2001 From: Hoang Bui <47828508+bongbui321@users.noreply.github.com> Date: Tue, 20 Aug 2024 14:06:18 -0400 Subject: [PATCH] Add done state: `on_broken_line_done` (#710) * add on_broken_line_done * fix --------- Co-authored-by: pengzhenghao --- metadrive/envs/metadrive_env.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/metadrive/envs/metadrive_env.py b/metadrive/envs/metadrive_env.py index 56446e4b5..29c445306 100644 --- a/metadrive/envs/metadrive_env.py +++ b/metadrive/envs/metadrive_env.py @@ -84,6 +84,7 @@ # ===== Termination Scheme ===== out_of_route_done=False, on_continuous_line_done=True, + on_broken_line_done=False, crash_vehicle_done=True, crash_object_done=True, crash_human_done=True, @@ -235,6 +236,8 @@ def _is_out_of_road(self, vehicle): ret = ret or vehicle.out_of_route elif self.config["on_continuous_line_done"]: ret = ret or vehicle.on_yellow_continuous_line or vehicle.on_white_continuous_line or vehicle.crash_sidewalk + if self.config["on_broken_line_done"]: + ret = ret or vehicle.on_broken_line return ret def reward_function(self, vehicle_id: str):