mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-08-07 15:15:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add type annotations to inputs and prepare_inputs method

Browse Source 
This commit adds explicit type annotations for the `inputs` dictionary and the return value of the `prepare_inputs` method in `model_runner.py`. These changes improve code readability and ensure type consistency, enhancing maintainability and reducing potential errors.
This commit is contained in:
DevTekVE
2024-12-29 17:50:58 +01:00
parent 6978e4faae
commit dffe04ad62
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
selfdrive/modeld/runners/model_runner.py
+2 -2
View File
@@ -31,7 +31,7 @@ class ModelRunner(ABC):
self.model_metadata = pickle.load(f)
self.input_shapes = self.model_metadata['input_shapes']
self.output_slices = self.model_metadata['output_slices']
self.inputs = {}
self.inputs: dict[str, np.ndarray | Tensor] = {}
@abstractmethod
def prepare_inputs(self, imgs_cl: dict[str, CLMem], numpy_inputs: dict[str, np.ndarray]) -> dict[str, np.ndarray]:
@@ -58,7 +58,7 @@ class TinygradRunner(ModelRunner):
with open(MODEL_PKL_PATH, "rb") as f:
self.model_run = pickle.load(f)
def prepare_inputs(self, imgs_cl: dict[str, CLMem], numpy_inputs: dict[str, np.ndarray]) -> dict[str, np.ndarray]:
def prepare_inputs(self, imgs_cl: dict[str, CLMem], numpy_inputs: dict[str, np.ndarray]) -> dict[str, Tensor]:
# Initialize image tensors if not already done
for key in imgs_cl:
if key not in self.inputs: