Fix formatting

Signed-off-by: Arthur Chan <[email protected]>
google · Dec 3, 2024 · 47cbbfc · 47cbbfc
1 parent 721bcb5
commit 47cbbfc
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/llm_toolkit/prompt_builder.py b/llm_toolkit/prompt_builder.py
@@ -1070,8 +1070,8 @@ def __init__(self,
     self.error_str = '\n'.join(errors)
 
     # Load templates.
-    self.template_file = self._find_template(template_dir,
-                                             'jvm_requirement_error_fixing.txt')
+    self.template_file = self._find_template(
+        template_dir, 'jvm_requirement_error_fixing.txt')
 
   def _find_template(self, template_dir: str, template_name: str) -> str:
     """Finds template file based on |template_dir|."""
@@ -1106,7 +1106,7 @@ def build(self,
 
     # Add the generated harness and error string to prompt
     prompt_text = prompt_text.replace('{GENERATED_HARNESS}',
-                                       self.generated_harness)
+                                      self.generated_harness)
     prompt_text = prompt_text.replace('{ERRORS}', self.error_str)
 
     self._prompt.add_priming(prompt_text)
@@ -1595,4 +1595,3 @@ def post_process_generated_code(self, generated_code: str) -> str:
       generated_code = generated_code.replace(
           'extern "C" int LLVMFuzzerTestOneInput', 'int LLVMFuzzerTestOneInput')
     return generated_code
-