make pre commit pass in the whole codebase (#149)

scripts/clean_benchmarks.py

@@ -2,26 +2,21 @@
 # for each folder, run the benchmark
 
 import os
-import sys
-import subprocess
-import time
-import datetime
 import shutil
-import argparse
-import json
-from pathlib import Path
-from typer import run
-from itertools import islice
 
-def main(
-):
-    benchmarks = Path('benchmark')
+from pathlib import Path
+
+from typer import run
+
+
+def main():
+    benchmarks = Path("benchmark")
 
     for benchmark in benchmarks.iterdir():
         if benchmark.is_dir():
-            print(f'Cleaning {benchmark}')
+            print(f"Cleaning {benchmark}")
             for path in benchmark.iterdir():
-                if path.name == 'main_prompt':
+                if path.name == "main_prompt":
                     continue
 
                 # Get filename of Path object
@@ -32,7 +27,6 @@ def main(
                     # delete the file
                     os.remove(path)
 
-if __name__ == '__main__':
-    run(main)
-            
 
+if __name__ == "__main__":
+    run(main)