Fix test (metadriverse#662)

* add more sanity check * fix test
commaai · Feb 24, 2024 · 65d17aa · 65d17aa
1 parent 905fbdd
commit 65d17aa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/metadrive/scenario/scenario_description.py b/metadrive/scenario/scenario_description.py
@@ -670,6 +670,7 @@ def offset_scenario_with_new_origin(scenario, new_origin):
 
         for light in scenario[ScenarioDescription.DYNAMIC_MAP_STATES].values():
             if ScenarioDescription.TRAFFIC_LIGHT_POSITION in light:
+                light["stop_point"] = np.asarray(light["stop_point"])
                 light[ScenarioDescription.TRAFFIC_LIGHT_POSITION][..., :2] -= new_origin
 
         scenario["metadata"]["old_origin_in_current_coordinate"] = -new_origin