linting

Author: KRRT7

codeflash/context/code_context_extractor.py

@@ -334,7 +334,9 @@ def extract_code_markdown_context_from_files(
                         helpers_of_fto.get(file_path, set()) | helpers_of_helpers.get(file_path, set())
                     ),
                 )
-            code_string_context = CodeString(code=code_context, file_path=file_path.resolve().relative_to(project_root_path.resolve()))
+            code_string_context = CodeString(
+                code=code_context, file_path=file_path.resolve().relative_to(project_root_path.resolve())
+            )
             code_context_markdown.code_strings.append(code_string_context)
     # Extract code from file paths containing helpers of helpers
     for file_path, helper_function_sources in helpers_of_helpers_no_overlap.items():
@@ -365,7 +367,9 @@ def extract_code_markdown_context_from_files(
                     project_root=project_root_path,
                     helper_functions=list(helpers_of_helpers_no_overlap.get(file_path, set())),
                 )
-            code_string_context = CodeString(code=code_context, file_path=file_path.resolve().relative_to(project_root_path.resolve()))
+            code_string_context = CodeString(
+                code=code_context, file_path=file_path.resolve().relative_to(project_root_path.resolve())
+            )
             code_context_markdown.code_strings.append(code_string_context)
     return code_context_markdown
 

codeflash/optimization/function_optimizer.py

@@ -1529,9 +1529,7 @@ def establish_original_code_baseline(
         test_env = self.get_test_env(codeflash_loop_index=0, codeflash_test_iteration=0, codeflash_tracer_disable=1)
 
         if self.function_to_optimize.is_async:
-            from codeflash.code_utils.instrument_existing_tests import (
-                instrument_source_module_with_async_decorators,
-            )
+            from codeflash.code_utils.instrument_existing_tests import instrument_source_module_with_async_decorators
 
             success, instrumented_source = instrument_source_module_with_async_decorators(
                 self.function_to_optimize.file_path, self.function_to_optimize, TestingMode.BEHAVIOR