Fixed some more ruff check issues

Author: zomglings

codeflash/code_utils/formatter.py

@@ -10,12 +10,12 @@
 
 from codeflash.cli_cmds.console import console, logger
 from codeflash.code_utils.code_replacer import OptimFunctionCollector
-from codeflash.discovery.functions_to_optimize import FunctionToOptimize
-from codeflash.models.models import FunctionParent, FunctionSource
 
 if TYPE_CHECKING:
     from pathlib import Path
 
+    from codeflash.discovery.functions_to_optimize import FunctionToOptimize
+    from codeflash.models.models import FunctionParent, FunctionSource
 
 def format_code(formatter_cmds: list[str], path: Path) -> str:
     # TODO: Only allow a particular whitelist of formatters here to prevent arbitrary code execution
@@ -66,7 +66,7 @@ def get_modification_code_ranges(
     preexisting_functions: set[tuple[str, tuple[FunctionParent,...]]],
     helper_functions: list[FunctionSource],
 ) -> list[tuple[int, int]]:
-    """Returns the starting and ending line numbers of modified and new functions in a file containing edits.
+    """Returns the starting and ending line numbers of modified and new functions in a file with edits.
     """
     modified_functions = set()
     modified_functions.add(fto.qualified_name)

codeflash/optimization/function_optimizer.py

@@ -633,8 +633,8 @@ def reformat_code_and_helpers(
                 raise Exception("Formatting had unexpected effects on code ranges")
 
             # It is important to sort in descending order so that the index arithmetic remains simple as we modify new_code
-            code_ranges_unformatted.sort(key=lambda range: range[0], reverse=True)
-            code_ranges_formatted.sort(key=lambda range: range[0], reverse=True)
+            code_ranges_unformatted.sort(key=lambda r: r[0], reverse=True)
+            code_ranges_formatted.sort(key=lambda r: r[0], reverse=True)
             formatted_code_lines = formatted_code.split("\n")
             new_code_lines = unformatted_code.split("\n")
             for range_0, range_1 in zip(code_ranges_unformatted, code_ranges_formatted):