From 4e4584bcc94ef42734271e37562acd02d3bb695c Mon Sep 17 00:00:00 2001
From: Yangge Li <li213@illinois.edu>
Date: Sat, 25 Mar 2023 16:49:15 -0500
Subject: [PATCH] small tweak on the examples

---
 demo/tacas2023/exp4/noisy_sensor.py |  4 ++--
 verse/analysis/verifier.py          |  3 +++
 verse/map/example_map/map_tacas.py  | 12 ++++++++++++
 3 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/demo/tacas2023/exp4/noisy_sensor.py b/demo/tacas2023/exp4/noisy_sensor.py
index 5dd6eb3b..890553a4 100644
--- a/demo/tacas2023/exp4/noisy_sensor.py
+++ b/demo/tacas2023/exp4/noisy_sensor.py
@@ -7,8 +7,8 @@ class NoisyVehicleSensor(BaseSensor):
         self.noise_x = noise_x
         self.noise_y = noise_y
 
-    def sense(self, scenario, agent, state_dict, track_map):
-        cont, disc, len_dict = super().sense(scenario, agent, state_dict, track_map)
+    def sense(self, agent, state_dict, track_map):
+        cont, disc, len_dict = super().sense(agent, state_dict, track_map)
         tmp = np.array(list(state_dict.values())[0][0])
         if tmp.ndim<2:
             return cont, disc, len_dict
diff --git a/verse/analysis/verifier.py b/verse/analysis/verifier.py
index 91dcf7cd..ea4b1371 100644
--- a/verse/analysis/verifier.py
+++ b/verse/analysis/verifier.py
@@ -433,6 +433,9 @@ class Verifier:
         past_runs,
         params = {},
     ):
+        if (max_height == None):
+            max_height = float('inf')
+        
         root = AnalysisTreeNode(
             trace={},
             init={},
diff --git a/verse/map/example_map/map_tacas.py b/verse/map/example_map/map_tacas.py
index c5ffbc1f..ea9285d9 100644
--- a/verse/map/example_map/map_tacas.py
+++ b/verse/map/example_map/map_tacas.py
@@ -254,9 +254,21 @@ class M5(LaneMap_3d):
         }
 
     def h(self, lane_idx: str, agent_mode_src: str, agent_mode_dest: str) -> str:
+        if not isinstance(lane_idx,str):
+            lane_idx = lane_idx.name
+        if not isinstance(agent_mode_src,str):
+            agent_mode_src = agent_mode_src.name
+        if not isinstance(agent_mode_dest,str):
+            agent_mode_dest = agent_mode_dest.name
         return self.h_dict[(lane_idx, agent_mode_src, agent_mode_dest)]
 
     def h_exist(self, lane_idx: str, agent_mode_src: str, agent_mode_dest: str) -> bool:
+        if not isinstance(lane_idx,str):
+            lane_idx = lane_idx.name
+        if not isinstance(agent_mode_src,str):
+            agent_mode_src = agent_mode_src.name
+        if not isinstance(agent_mode_dest,str):
+            agent_mode_dest = agent_mode_dest.name
         if (lane_idx, agent_mode_src, agent_mode_dest) in self.h_dict:
             return True
         else:
-- 
GitLab