Format bench-TriLMs.py with black

bench-TriLMs.py

@@ -117,7 +117,7 @@ def llama_bench(
                 logger.debug(result.stderr.decode(errors="ignore"))
                 if result.returncode != 0 or len(result.stdout) == 0:
                     logger.error("Failed to run %s", " ".join(command))
-                    break;
+                    break
 
                 new_output = json.loads(result.stdout)
                 logger.info(json.dumps(new_output, indent=4))
@@ -171,7 +171,9 @@ 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")
+    parser.add_argument(
+        "--force", action="store_true", help="Overwrite the result file without asking"
+    )
     return parser.parse_args(args[1:])
 
 
@@ -216,13 +218,15 @@ if __name__ == "__main__":
 
     if shutil.which("lscpu") is not None:
         logger.info("Getting CPU info")
-        final_result["cpuinfo"] = subprocess.run(["lscpu"], capture_output=True).stdout.decode(
-            encoding="utf-8"
-        )
+        final_result["cpuinfo"] = subprocess.run(
+            ["lscpu"], capture_output=True
+        ).stdout.decode(encoding="utf-8")
 
     if args.gpu and shutil.which("nvidia-smi") is not None:
         logger.info("Getting NVIDIA GPU info")
-        final_result["gpuinfo"] = subprocess.run(["nvidia-smi", "-q"], capture_output=True).stdout.decode(encoding="utf-8")
+        final_result["gpuinfo"] = subprocess.run(
+            ["nvidia-smi", "-q"], capture_output=True
+        ).stdout.decode(encoding="utf-8")
 
     logger.info("Writing output to: %s", output_file)
     logger.debug("Final results: %s", json.dumps(final_result, indent=4))