feedback, cleanup

Author: aseembits93

.idea/runConfigurations/retriever_local.xml

@@ -1,11 +1,18 @@
-from code_to_optimize.bubble_sort_dep1_helper import dep1_comparer
-from code_to_optimize.bubble_sort_dep2_swap import dep2_swap
 
 
 def sorter_deps(arr):
-    for i in range(len(arr)):
-        for j in range(len(arr) - 1):
-            if dep1_comparer(arr, j):
-                dep2_swap(arr, j)
+    n = len(arr)
+    for i in range(n):
+        # Introduce a flag that will allow us to exit early if no swaps occur
+        swapped = False
+        # Only iterate to the unsorted portion of the list
+        for j in range(n - 1 - i):
+            # Inline the comparison function
+            if arr[j] > arr[j + 1]:
+                # Inline the swap
+                arr[j], arr[j + 1] = arr[j + 1], arr[j]
+                swapped = True
+        # If no swaps occurred in the last inner loop, the array is sorted
+        if not swapped:
+            break
     return arr
-

code_to_optimize/bubble_sort_deps.py

@@ -10,7 +10,6 @@
 from codeflash.cli_cmds.console import logger
 from codeflash.code_utils.code_extractor import add_needed_imports_from_module
 from codeflash.models.models import FunctionParent
-import isort
 
 if TYPE_CHECKING:
     from pathlib import Path
@@ -336,87 +335,4 @@ def function_to_optimize_original_worktree_fqn(
         str(worktrees[0].name / function_to_optimize.file_path.relative_to(git_root).with_suffix("")).replace("/", ".")
         + "."
         + function_to_optimize.qualified_name
-    )
-
-
-def add_decorator_cst(module_node, function_name, decorator_name):
-    """Adds a decorator to a function definition in a LibCST module node."""
-
-    class AddDecoratorTransformer(cst.CSTTransformer):
-        def leave_FunctionDef(self, original_node, updated_node):
-            if original_node.name.value == function_name:
-                new_decorator = cst.Decorator(
-                    decorator=cst.Name(value=decorator_name)
-                )
-
-                updated_decorators = list(updated_node.decorators)
-                updated_decorators.insert(0, new_decorator)
-
-                return updated_node.with_changes(decorators=updated_decorators)
-            return updated_node
-
-    transformer = AddDecoratorTransformer()
-    updated_module = module_node.visit(transformer)
-    return updated_module
-
-def add_decorator_imports(file_paths, fn_list, db_file):
-    """Adds a decorator to a function in a Python file."""
-    for file_path, fn_name in zip(file_paths, fn_list):
-        #open file
-        with open(file_path, "r", encoding="utf-8") as file:
-            file_contents = file.read()
-
-        # parse to cst
-        module_node = cst.parse_module(file_contents)
-        # add decorator
-        module_node = add_decorator_cst(module_node, fn_name, 'profile')
-        # add imports
-        # Create a transformer to add the import
-        transformer = ImportAdder("from line_profiler import profile")
-
-        # Apply the transformer to add the import
-        module_node = module_node.visit(transformer)
-        modified_code = isort.code(module_node.code, float_to_top=True)
-        # write to file
-        with open(file_path, "w", encoding="utf-8") as file:
-            file.write(modified_code)
-    #do this only for the main file and not the helper files, can use libcst but will go just with some simple string manipulation
-    with open(file_paths[0],'r') as f:
-        file_contents = f.readlines()
-    for idx, line in enumerate(file_contents):
-        if 'from line_profiler import profile' in line:
-            file_contents.insert(idx+1, f"profile.enable(output_prefix='{db_file}')\n")
-            break
-    with open(file_paths[0],'w') as f:
-        f.writelines(file_contents)
-
-
-
-
-class ImportAdder(cst.CSTTransformer):
-    def __init__(self, import_statement='from line_profiler import profile'):
-        self.import_statement = import_statement
-        self.has_import = False
-
-    def leave_Module(self, original_node, updated_node):
-        # If the import is already there, don't add it again
-        if self.has_import:
-            return updated_node
-
-        # Parse the import statement into a CST node
-        import_node = cst.parse_statement(self.import_statement)
-
-        # Add the import to the module's body
-        return updated_node.with_changes(
-            body=[import_node] + list(updated_node.body)
-        )
-
-    def visit_Import(self, node):
-        pass
-
-    def visit_ImportFrom(self, node):
-        # Check if the profile is already imported from line_profiler
-        if node.module and node.module.value == "line_profiler":
-            for import_alias in node.names:
-                if import_alias.name.value == "profile":
-                    self.has_import = True
+    )

code_to_optimize/retriever_bubble_sort_imported.py

@@ -59,8 +59,3 @@ def prepare_coverage_files() -> tuple[Path, Path]:
     coveragerc_content = f"[run]\n branch = True\ndata_file={coverage_database_file}\n"
     coveragercfile.write_text(coveragerc_content)
     return coverage_database_file, coveragercfile
-
-def prepare_lprofiler_files(prefix="") -> tuple[Path]:
-    """Prepare line profiler output file."""
-    lprofiler_database_file = get_run_tmp_file(prefix)
-    return lprofiler_database_file