Fix the missing new line issue of dynamic shape.

kendryte · Feb 21, 2024 · cc9c565 · cc9c565
1 parent 699f198
commit cc9c565
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/inference.py b/tests/inference.py
@@ -99,7 +99,7 @@ def set_infer_input(self, sim, compile_opt):
     def dump_kmodel_desc(self, file):
         input_shapes = data_shape_list_string(self.inputs)
         output_shapes = data_shape_list_string(self.outputs)
-        s = f"{len(self.inputs)} {len(self.outputs)}\n{input_shapes}\n{output_shapes}"
+        s = f"{len(self.inputs)} {len(self.outputs)}\n{input_shapes}\n{output_shapes}\n"
         with open(file, "w+") as f:
             f.write(s)