diff --git a/mlonmcu/models/frontend.py b/mlonmcu/models/frontend.py index f325dd816..0a425e61e 100644 --- a/mlonmcu/models/frontend.py +++ b/mlonmcu/models/frontend.py @@ -249,7 +249,7 @@ def produce_artifacts(self, model): ext = self.input_formats[0].extension with open(path, "rb") as handle: # TODO: is an onnx model raw data or text? raw = handle.read() - artifacts.append(Artifact(f"{name}.{ext}", raw=raw, fmt=ArtifactFormat.RAW), flags=["model"]) + artifacts.append(Artifact(f"{name}.{ext}", raw=raw, fmt=ArtifactFormat.RAW, flags=["model"])) return artifacts @@ -304,7 +304,7 @@ def produce_artifacts(self, model): ext = self.input_formats[0].extension with open(path, "rb") as handle: raw = handle.read() - artifacts.append(Artifact(f"{name}.{ext}", raw=raw, fmt=ArtifactFormat.RAW), flags=["model"]) + artifacts.append(Artifact(f"{name}.{ext}", raw=raw, fmt=ArtifactFormat.RAW, flags=["model"])) if not self.visualize_enable: assert len(self.output_formats) == 1 @@ -431,7 +431,7 @@ def produce_artifacts(self, model): ext = self.input_formats[0].extension with open(path, "rb") as handle: # TODO: is an onnx model raw data or text? raw = handle.read() - artifacts.append(Artifact(f"{name}.{ext}", raw=raw, fmt=ArtifactFormat.RAW), flags=["model"]) + artifacts.append(Artifact(f"{name}.{ext}", raw=raw, fmt=ArtifactFormat.RAW, flags=["model"])) if not self.visualize_graph: assert len(self.output_formats) == 1