From d237c175cdccda59e9db4351e9d1e66a24f5536d Mon Sep 17 00:00:00 2001
From: Junlin Liu <liu@junl.in>
Date: Sun, 18 Jun 2023 21:36:45 +0800
Subject: [PATCH] fix docstring #150 (#152)

---
 gpt_engineer/main.py |  5 ++++-
 identity/generate    |  2 +-
 scripts/benchmark.py | 14 +++++++++++---
 3 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/gpt_engineer/main.py b/gpt_engineer/main.py
index e0dff9c..059f754 100644
--- a/gpt_engineer/main.py
+++ b/gpt_engineer/main.py
@@ -18,7 +18,10 @@ def chat(
     delete_existing: str = typer.Argument(None, help="delete existing files"),
     run_prefix: str = typer.Option(
         "",
-        help="run prefix, if you want to run multiple variants of the same project and later compare them",
+        help=(
+            "run prefix, if you want to run multiple variants of the same project and "
+            "later compare them"
+        ),
     ),
     model: str = "gpt-4",
     temperature: float = 0.1,
diff --git a/identity/generate b/identity/generate
index 401934e..1c1ac62 100644
--- a/identity/generate
+++ b/identity/generate
@@ -4,7 +4,7 @@ Make sure that every detail of the architecture is, in the end, implemented as c
 
 You will first lay out the names of the core classes, functions, methods that will be necessary, as well as a quick comment on their purpose.
 Before you start outputting the code, you will output a seperator in the form of a line containing "*CODEBLOCKSBELOW*"
-Make sure to create any appropriate module dependency or package manager dependency definition file. 
+Make sure to create any appropriate module dependency or package manager dependency definition file.
 Then you will format and output the content, including ALL code, of each file strictly following a markdown code block format, where the following tokens should be replaced such that [FILENAME] is the lowercase file name including the file extension, [LANG] is the markup code block language for the code's language, and [CODE] is the code:
 [FILENAME]
 ```[LANG]
diff --git a/scripts/benchmark.py b/scripts/benchmark.py
index 3ff4b23..c07c9f2 100644
--- a/scripts/benchmark.py
+++ b/scripts/benchmark.py
@@ -30,7 +30,7 @@ def main(
             process = subprocess.Popen(
                 [
                     "python",
-                    "-u", # Unbuffered output
+                    "-u",  # Unbuffered output
                     "-m",
                     "gpt_engineer.main",
                     bench_folder,
@@ -50,7 +50,7 @@ def main(
         file.close()
 
         print("process", bench_folder.name, "finished with code", process.returncode)
-        print('Running it. Original benchmark prompt:')
+        print("Running it. Original benchmark prompt:")
         print()
         with open(bench_folder / "main_prompt") as f:
             print(f.read())
@@ -58,10 +58,18 @@ def main(
 
         try:
             subprocess.run(
-                ['python', "-m", "gpt_engineer.main", bench_folder, "--steps-config", "execute_only"],
+                [
+                    "python",
+                    "-m",
+                    "gpt_engineer.main",
+                    bench_folder,
+                    "--steps-config",
+                    "execute_only",
+                ],
             )
         except KeyboardInterrupt:
             pass
 
+
 if __name__ == "__main__":
     run(main)