Set default contact result to unset (#546)

default contact unset
metadriverse · Nov 5, 2023 · 0d43709 · 0d43709
1 parent 62353d6
commit 0d43709
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/metadrive/constants.py b/metadrive/constants.py
@@ -81,6 +81,7 @@ class TerminationState:
     MetaDriveType.LIGHT_RED: "red",
     MetaDriveType.LIGHT_YELLOW: "orange",
     MetaDriveType.LIGHT_GREEN: "green",
+    MetaDriveType.UNSET: "orange",
 }
 
 

diff --git a/metadrive/engine/interface.py b/metadrive/engine/interface.py
@@ -158,7 +158,7 @@ def _render_banner(self, text, color=COLLISION_INFO_COLOR["green"][1]):
 
     def _render_contact_result(self, contacts):
         contacts = sorted(list(contacts), key=lambda c: COLLISION_INFO_COLOR[COLOR[c]][0])
-        text = contacts[0] if len(contacts) != 0 else None
+        text = contacts[0] if len(contacts) != 0 else MetaDriveType.UNSET
         color = COLLISION_INFO_COLOR[COLOR[text]][1]
         if time.time() - self.engine._episode_start_time < 10:
             text = "Press H to see help message"