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

…its target_speed Fix #563
Farama-Foundation · Jan 27, 2024 · 6577ef6 · 6577ef6
1 parent dcdfef3
commit 6577ef6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/highway_env/vehicle/behavior.py b/highway_env/vehicle/behavior.py
@@ -451,7 +451,10 @@ def acceleration_features(
     ) -> np.ndarray:
         vt, dv, dp = 0, 0, 0
         if ego_vehicle:
-            vt = getattr(ego_vehicle, "target_speed", ego_vehicle.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