From 6978e4faae92c3f0c1a79a760a41afdc4789e9a0 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)