frontend.py: fix

tum-ei-eda · Nov 1, 2023 · dd00c6e · dd00c6e
1 parent c317af1
commit dd00c6e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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