Ask before overwriting the output file

bench-TriLMs.py

@@ -170,6 +170,7 @@ def parse_args(args: Sequence[str]):
         default=Path(os.path.curdir) / "result.json",
         help="Path of the benchmark results to be written",
     )
+    parser.add_argument("--force", action="store_true", help="Overwrite the result file without asking")
     return parser.parse_args(args[1:])
 
 
@@ -181,6 +182,14 @@ if __name__ == "__main__":
     LLAMA_CPP_PATH = args.llama_cpp_path
     MODEL_DIR = args.model_dir
 
+    output_file = Path(args.out).absolute()
+
+    if output_file.exists() and not args.force:
+        ask = input("Result file exists. Do you want to overwrite it? [y/N]")
+        if not ask.strip().lower().startswith("y"):
+            logger.info("Not running, leaving output file intact")
+            exit()
+
     results = []
     mulmat_perf = []
     repetitions: int = args.repetitions
@@ -208,8 +217,8 @@ if __name__ == "__main__":
         "mulmat_perf": mulmat_perf,
         "results": results,
     }
-    logger.info("Writing output to: %s", args.out)
+    logger.info("Writing output to: %s", output_file)
     logger.debug("Final results: %s", json.dumps(final_result, indent=4))
-    with open(args.out, "w") as f:
+    with open(output_file, "w") as f:
         json.dump(final_result, f, indent=4)
         f.flush()