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 550c08ac4c
commit c39f722f7b
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)