Merge pull request #568 from Farama-Foundation/fix-563

Fix mobil's prediction of a vehicle's behaviour without knowledge of …
Farama-Foundation · Jan 27, 2024 · 81c11d6 · 81c11d6
2 parents 17059b8 + 6577ef6
commit 81c11d6
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 = 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