From 1b419974b04fdb25e15b45cea38cf5cf7e090c72 Mon Sep 17 00:00:00 2001 From: Jason Wen Date: Sat, 13 Jul 2024 11:52:44 -0400 Subject: [PATCH] Revert "fix types" This reverts commit 3cdbde6dbe5871796bade3e67031495267c860f8. --- selfdrive/modeld/custom_model_metadata.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/selfdrive/modeld/custom_model_metadata.py b/selfdrive/modeld/custom_model_metadata.py index 783d11be22..685d9655e4 100644 --- a/selfdrive/modeld/custom_model_metadata.py +++ b/selfdrive/modeld/custom_model_metadata.py @@ -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