Update code_extractor.py

Author: KRRT7

codeflash/code_utils/code_extractor.py

@@ -1,3 +1,4 @@
+# ruff: noqa: ARG002
 from __future__ import annotations
 
 import ast
@@ -124,10 +125,11 @@ def leave_Module(self, original_node: cst.Module, updated_node: cst.Module) -> c
         new_statements = list(updated_node.body)
 
         # Find assignments to append
-        assignments_to_append = []
-        for name in self.new_assignment_order:
-            if name not in self.processed_assignments and name in self.new_assignments:
-                assignments_to_append.append(self.new_assignments[name])
+        assignments_to_append = [
+            self.new_assignments[name]
+            for name in self.new_assignment_order
+            if name not in self.processed_assignments and name in self.new_assignments
+        ]
 
         if assignments_to_append:
             # Add a blank line before appending new assignments if needed
@@ -136,8 +138,12 @@ def leave_Module(self, original_node: cst.Module, updated_node: cst.Module) -> c
                 new_statements.pop()  # Remove the Pass statement but keep the empty line
 
             # Add the new assignments
-            for assignment in assignments_to_append:
-                new_statements.append(cst.SimpleStatementLine([assignment], leading_lines=[cst.EmptyLine()]))
+            new_statements.extend(
+                [
+                    cst.SimpleStatementLine([assignment], leading_lines=[cst.EmptyLine()])
+                    for assignment in assignments_to_append
+                ]
+            )
 
         return updated_node.with_changes(body=new_statements)
 
@@ -426,7 +432,7 @@ def find_target(node_list: list[ast.stmt], name_parts: tuple[str, str] | tuple[s
 
         return find_target(target.body, name_parts[1:])
 
-    with open(file_path, encoding="utf8") as file:
+    with file_path.open(encoding="utf8") as file:
         source_code: str = file.read()
     try:
         module_node: ast.Module = ast.parse(source_code)