Fix mobil's prediction of a vehicle's behaviour without knowledge of …

…its target_speed Fix #563
Farama-Foundation · Jan 27, 2024 · dcdfef3 · dcdfef3
1 parent 17059b8
commit dcdfef3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/highway_env/vehicle/behavior.py b/highway_env/vehicle/behavior.py
@@ -451,7 +451,7 @@ def acceleration_features(
     ) -> np.ndarray:
         vt, dv, dp = 0, 0, 0
         if ego_vehicle:
-            vt = ego_vehicle.target_speed - ego_vehicle.speed
+            vt = getattr(ego_vehicle, "target_speed", ego_vehicle.speed) - ego_vehicle.speed
             d_safe = (
                 self.DISTANCE_WANTED
                 + np.maximum(ego_vehicle.speed, 0) * self.TIME_WANTED