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.
This commit is contained in:
DevTekVE
2024-12-29 17:48:00 +01:00
parent 9ec42cd6ff
commit 6978e4faae
2 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -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]
+3 -3
View File
@@ -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)