From 65d17aa7837e3425ba495d68ef9ffda3e36a715d Mon Sep 17 00:00:00 2001 From: Quanyi Li Date: Sat, 24 Feb 2024 14:56:12 +0000 Subject: [PATCH] Fix test (#662) * add more sanity check * fix test --- metadrive/scenario/scenario_description.py | 1 + 1 file changed, 1 insertion(+) diff --git a/metadrive/scenario/scenario_description.py b/metadrive/scenario/scenario_description.py index fd119a04d..a3a3103ca 100644 --- 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