diff --git a/backends/arm/arm_vela.py b/backends/arm/arm_vela.py index ef0336147af..f7f0c4b49cc 100644 --- a/backends/arm/arm_vela.py +++ b/backends/arm/arm_vela.py @@ -96,13 +96,13 @@ def vela_compile(tosa_graph, args: List[str], shape_order=None): block_name = block_name + b"\x00" * (16 - len(block_name)) # We need the acual unpadded block lengths for hw setup - block_length = struct.pack(" defaultdict: """ @@ -104,7 +104,7 @@ def get_compression_ratio(self) -> float: return compression_ratio - def evaluate(self) -> dict[Any]: # type: ignore[type-arg] + def evaluate(self) -> dict[str, Any]: model_error_dict = self.get_model_error() output_metrics = {"name": self.model_name, "metrics": dict(model_error_dict)}