formatting

Author: KRRT7

codeflash/code_utils/code_extractor.py

@@ -1131,6 +1131,7 @@ def find_specific_function_in_file(
     """
     try:
         from codeflash.code_utils.compat import get_jedi_environment
+
         jedi_env = get_jedi_environment()
         logger.debug(f"find_specific_function_in_file using Jedi environment: {jedi_env}")
 
@@ -1165,14 +1166,12 @@ def get_fn_references_jedi(
     )
     try:
         from codeflash.code_utils.compat import get_jedi_environment
+
         jedi_env = get_jedi_environment()
         logger.debug(f"get_fn_references_jedi using Jedi environment: {jedi_env}")
 
         script = jedi.Script(
-            code=source_code,
-            path=file_path,
-            project=jedi.Project(path=project_root),
-            environment=jedi_env
+            code=source_code, path=file_path, project=jedi.Project(path=project_root), environment=jedi_env
         )
         # Get references to the function
         references = script.get_references(line=function_position.line_no, column=function_position.col_no)

codeflash/code_utils/compat.py

@@ -141,5 +141,6 @@ def get_jedi_environment() -> InterpreterEnvironment | None:
         return InterpreterEnvironment()
     except Exception as e:
         from codeflash.cli_cmds.console import logger
+
         logger.warning(f"Could not create InterpreterEnvironment: {e}")
         return None

codeflash/context/code_context_extractor.py

@@ -422,9 +422,7 @@ def get_function_to_optimize_as_function_source(
         logger.warning(f"Using Jedi environment: {jedi_env}")
 
         script = jedi.Script(
-            path=function_to_optimize.file_path,
-            project=jedi.Project(path=project_root_path),
-            environment=jedi_env
+            path=function_to_optimize.file_path, project=jedi.Project(path=project_root_path), environment=jedi_env
         )
 
         # Get all names in the file
@@ -476,11 +474,7 @@ def get_function_sources_from_jedi(
 
     for file_path, qualified_function_names in file_path_to_qualified_function_names.items():
         try:
-            script = jedi.Script(
-                path=file_path,
-                project=jedi.Project(path=project_root_path),
-                environment=jedi_env
-            )
+            script = jedi.Script(path=file_path, project=jedi.Project(path=project_root_path), environment=jedi_env)
             file_refs = script.get_names(all_scopes=True, definitions=False, references=True)
 
             for qualified_function_name in qualified_function_names:

codeflash/discovery/discover_unit_tests.py

@@ -675,13 +675,7 @@ def discover_tests_pytest(
             cwd=project_root, check=False, text=True, capture_output=True
         )
         result = subprocess.run(  # noqa: PLW1510
-            [
-                SAFE_SYS_EXECUTABLE,
-                str(discovery_script_path),
-                str(project_root),
-                str(tests_root),
-                str(tmp_pickle_path),
-            ],
+            [SAFE_SYS_EXECUTABLE, str(discovery_script_path), str(project_root), str(tests_root), str(tmp_pickle_path)],
             **run_kwargs,
         )
     try:
@@ -863,6 +857,7 @@ def process_test_files(
                 continue
             try:
                 from codeflash.code_utils.compat import get_jedi_environment
+
                 jedi_env = get_jedi_environment()
                 logger.debug(f"process_test_files using Jedi environment: {jedi_env}")
 
@@ -931,7 +926,9 @@ def process_test_files(
                                 and f".{matched_name}." in def_name.full_name
                             ):
                                 for function in functions_to_search:
-                                    (is_parameterized, new_function, parameters) = discover_parameters_unittest(function)
+                                    (is_parameterized, new_function, parameters) = discover_parameters_unittest(
+                                        function
+                                    )
 
                                     if is_parameterized and new_function == def_name.name:
                                         test_functions.add(