From c39f722f7bf78ec7e01f4cda1d6dd13428243077 Mon Sep 17 00:00:00 2001 From: DevTekVE Date: Sun, 29 Dec 2024 17:48:00 +0100 Subject: [PATCH] Rename TinyGradRunner to TinygradRunner for consistency. The class name and references were updated to maintain naming consistency across the codebase. This aligns with naming conventions and improves code clarity. No functional changes were introduced. --- selfdrive/modeld/modeld.py | 4 ++-- selfdrive/modeld/runners/model_runner.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/selfdrive/modeld/modeld.py b/selfdrive/modeld/modeld.py index 3d6341dc29..8e26b78e27 100755 --- a/selfdrive/modeld/modeld.py +++ b/selfdrive/modeld/modeld.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 from openpilot.system.hardware import TICI -from openpilot.selfdrive.modeld.runners.model_runner import ONNXRunner, TinyGradRunner +from openpilot.selfdrive.modeld.runners.model_runner import ONNXRunner, TinygradRunner # import time @@ -57,7 +57,7 @@ class ModelState: } # Initialize model runner - self.model_runner = TinyGradRunner() if TICI else ONNXRunner(self.frames) + self.model_runner = TinygradRunner() if TICI else ONNXRunner(self.frames) self.parser = Parser() net_output_size = self.model_runner.model_metadata['output_shapes']['outputs'][1] diff --git a/selfdrive/modeld/runners/model_runner.py b/selfdrive/modeld/runners/model_runner.py index 77d1c97ba1..53da7cfd57 100644 --- a/selfdrive/modeld/runners/model_runner.py +++ b/selfdrive/modeld/runners/model_runner.py @@ -49,12 +49,12 @@ class ModelRunner(ABC): return parsed_outputs -class TinyGradRunner(ModelRunner): - """TinyGrad implementation of model runner for TICI hardware.""" +class TinygradRunner(ModelRunner): + """Tinygrad implementation of model runner for TICI hardware.""" def __init__(self): super().__init__() - # Load TinyGrad model + # Load Tinygrad model with open(MODEL_PKL_PATH, "rb") as f: self.model_run = pickle.load(f)