mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-09-27 19:43:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

enforce type hint

Browse Source
This commit is contained in:
Jason Wen
2024-07-01 11:58:11 -04:00
parent 6e6de6742e
commit 7d751da1b8
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
selfdrive/controls/lib/sunnypilot/accel_controller.py
+4 -3
View File
@@ -52,13 +52,13 @@ class AccelController:
def __init__(self):
self._profile = DPAccel.STOCK
def set_profile(self, profile):
def set_profile(self, profile: int):
try:
self._profile = profile if profile in DPAccel.accel_val() else DPAccel.STOCK
except (ValueError, TypeError):
self._profile = DPAccel.STOCK
def _dp_calc_cruise_accel_limits(self, v_ego):
def _dp_calc_cruise_accel_limits(self, v_ego: float):
if self._profile == DPAccel.ECO:
min_v = _DP_CRUISE_MIN_V_ECO
max_v = _DP_CRUISE_MAX_V_ECO
@@ -71,9 +71,10 @@ class AccelController:
a_cruise_min = interp(v_ego, _DP_CRUISE_MIN_BP, min_v)
a_cruise_max = interp(v_ego, _DP_CRUISE_MAX_BP, max_v)
return a_cruise_min, a_cruise_max
def get_accel_limits(self, v_ego, accel_limits):
def get_accel_limits(self, v_ego: float, accel_limits: list[float]):
return accel_limits if self._profile == DPAccel.STOCK else self._dp_calc_cruise_accel_limits(v_ego)
def is_enabled(self):