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

Revert "fix types"

Browse Source 
This reverts commit 3cdbde6dbe.
This commit is contained in:
Jason Wen
2024-07-13 11:52:44 -04:00
parent 3cdbde6dbe
commit 1b419974b0
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
selfdrive/modeld/custom_model_metadata.py
+3 -3
View File
@@ -48,14 +48,14 @@ class CustomModelMetadata:
self.params: Params = params
self.generation: ModelGeneration = self.read_model_generation_param()
self.capabilities: ModelCapabilities = self.get_model_capabilities()
self.capabilities: int = self.get_model_capabilities()
self.valid: bool = self.params.get_bool("CustomDrivingModel") and not SIMULATION and \
self.capabilities != ModelCapabilities.Default
def read_model_generation_param(self) -> ModelGeneration:
return ModelGeneration(int(self.params.get("DrivingModelGeneration", encoding='utf8')) or ModelGeneration.DEFAULT)
return int(self.params.get('DrivingModelGeneration') or ModelGeneration.DEFAULT)
def get_model_capabilities(self) -> ModelCapabilities:
def get_model_capabilities(self) -> int:
"""Returns the model capabilities for a given generation."""
if self.generation == ModelGeneration.FIVE:
return ModelCapabilities.DesiredCurvatureV2