Add done state: on_broken_line_done (#710)

* add on_broken_line_done * fix --------- Co-authored-by: pengzhenghao <[email protected]>
metadriverse · Aug 20, 2024 · 880fcd0 · 880fcd0
1 parent aaed1f7
commit 880fcd0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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):