diff --git a/sunnypilot/mads/mads.py b/sunnypilot/mads/mads.py index 6fd1c9086f..1857505e47 100644 --- a/sunnypilot/mads/mads.py +++ b/sunnypilot/mads/mads.py @@ -26,14 +26,13 @@ class ModifiedAssistDrivingSystem: self.disengage_lateral_on_brake_toggle = mads_params.read_param("MadsDisengageLateralOnBrake", self.selfdrive.params) self.unified_engagement_mode = mads_params.read_param("MadsUnifiedEngagementMode", self.selfdrive.params) - def update_availability(self, CS: car.CarState, available: bool = False) -> bool: + def update_availability(self, CS: car.CarState) -> None: if self.main_enabled_toggle: - available |= CS.cruiseState.available + self.available = CS.cruiseState.available if self.selfdrive.CP.carName == "hyundai": - available = True - - return available + if any(be.type == ButtonType.lkas and be.pressed for be in CS.buttonEvents): + self.available = True def update_events(self, CS: car.CarState): self.selfdrive.events.remove(EventName.pcmDisable) @@ -86,7 +85,7 @@ class ModifiedAssistDrivingSystem: if not self.enabled_toggle: return - self.available = self.update_availability(CS) + self.update_availability(CS) self.update_events(CS)