go

Author: KRRT7

codeflash/cli_cmds/cmd_init.py

@@ -963,4 +963,3 @@ def ask_for_telemetry() -> bool:
         default=True,
         show_default=True,
     )
-

codeflash/code_utils/code_extractor.py

@@ -21,7 +21,6 @@
     from codeflash.models.models import FunctionSource
 
 
-
 class GlobalAssignmentCollector(cst.CSTVisitor):
     """Collects all global assignment statements."""
 

codeflash/context/code_context_extractor.py

@@ -138,7 +138,7 @@ def extract_code_string_context_from_files(
     helpers_of_fto: dict[Path, set[FunctionSource]],
     helpers_of_helpers: dict[Path, set[FunctionSource]],
     project_root_path: Path,
-    remove_docstrings: bool = False,
+    remove_docstrings: bool = False,  # noqa: FBT001, FBT002
     code_context_type: CodeContextType = CodeContextType.READ_ONLY,
 ) -> CodeString:
     """Extract code context from files containing target functions and their helpers.
@@ -245,7 +245,7 @@ def extract_code_markdown_context_from_files(
     helpers_of_fto: dict[Path, set[FunctionSource]],
     helpers_of_helpers: dict[Path, set[FunctionSource]],
     project_root_path: Path,
-    remove_docstrings: bool = False,
+    remove_docstrings: bool = False,  # noqa: FBT001, FBT002
     code_context_type: CodeContextType = CodeContextType.READ_ONLY,
 ) -> CodeStringsMarkdown:
     """Extract code context from files containing target functions and their helpers, formatting them as markdown.
@@ -271,12 +271,12 @@ def extract_code_markdown_context_from_files(
     """
     # Rearrange to remove overlaps, so we only access each file path once
     helpers_of_helpers_no_overlap = defaultdict(set)
-    for file_path in helpers_of_helpers:
+    for file_path, helper_functions in helpers_of_helpers.items():
         if file_path in helpers_of_fto:
             # Remove duplicates within the same file path, in case a helper of helper is also a helper of fto
             helpers_of_helpers[file_path] -= helpers_of_fto[file_path]
         else:
-            helpers_of_helpers_no_overlap[file_path] = helpers_of_helpers[file_path]
+            helpers_of_helpers_no_overlap[file_path] = helper_functions
     code_context_markdown = CodeStringsMarkdown()
     # Extract code from file paths that contain fto and first degree helpers. helpers of helpers may also be included if they are in the same files
     for file_path, function_sources in helpers_of_fto.items():

codeflash/optimization/function_optimizer.py

@@ -404,7 +404,7 @@ def determine_best_candidate(
                         future_line_profile_results = None
                     try:
                         candidate = candidates.popleft()
-                    except IndexError as e:
+                    except IndexError:
                         if done:
                             break
                         continue