Update printer.print calls to match expected args (#779)

nod-ai · Jul 16, 2024 · 26ce08e · 26ce08e
1 parent d24df8f
commit 26ce08e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/turbine_models/custom_models/pipeline_base.py b/models/turbine_models/custom_models/pipeline_base.py
@@ -452,7 +452,7 @@ def prepare_all(
                         exit()
                 for submodel in self.map.keys():
                     if not self.map[submodel].get("vmfb"):
-                        self.printer.print("Fetching: ", submodel)
+                        self.printer.print(f"Fetching: {submodel}")
                         self.export_submodel(
                             submodel, input_mlir=self.map[submodel].get("mlir")
                         )
@@ -557,7 +557,7 @@ def is_prepared(self, vmfbs, weights):
         if not any(x for x in missing.values()):
             ready = True
         else:
-            self.printer.print("Missing files: ", missing)
+            self.printer.print(f"Missing files: {missing}")
             ready = False
         return ready