Merge branch 'main' into trace-and-optimize

Author: KRRT7

codeflash/cli_cmds/cmd_init.py

@@ -231,8 +231,7 @@ def collect_setup_info() -> SetupInfo:
     info_panel = Panel(
         Text(
             "📁 Let's identify your Python module directory.\n\n"
-            "This is usually the top-level directory containing all your Python source code.\n"
-            "We've automatically detected some directories for you.",
+            "This is usually the top-level directory containing all your Python source code.\n",
             style="cyan",
         ),
         title="🔍 Module Discovery",
@@ -936,7 +935,6 @@ def configure_pyproject_toml(setup_info: SetupInfo) -> None:
     codeflash_section["module-root"] = setup_info.module_root
     codeflash_section["tests-root"] = setup_info.tests_root
     codeflash_section["test-framework"] = setup_info.test_framework
-    codeflash_section["benchmarks-root"] = setup_info.benchmarks_root if setup_info.benchmarks_root else ""
     codeflash_section["ignore-paths"] = setup_info.ignore_paths
     codeflash_section["disable-telemetry"] = not enable_telemetry
     if setup_info.git_remote not in ["", "origin"]:

codeflash/discovery/functions_to_optimize.py

@@ -30,6 +30,8 @@
 from codeflash.telemetry.posthog_cf import ph
 
 if TYPE_CHECKING:
+    from argparse import Namespace
+
     from libcst import CSTNode
     from libcst.metadata import CodeRange
 
@@ -452,7 +454,7 @@ def inspect_top_level_functions_or_methods(
 
 
 def was_function_previously_optimized(
-    function_to_optimize: FunctionToOptimize, code_context: CodeOptimizationContext
+    function_to_optimize: FunctionToOptimize, code_context: CodeOptimizationContext, args: Namespace
 ) -> bool:
     """Check which functions have already been optimized and filter them out.
 
@@ -474,7 +476,7 @@ def was_function_previously_optimized(
         owner, repo = None, None
     pr_number = get_pr_number()
 
-    if not owner or not repo or pr_number is None:
+    if not owner or not repo or pr_number is None or getattr(args, "no_pr", False):
         return False
 
     code_contexts = []

codeflash/optimization/function_optimizer.py

@@ -170,7 +170,7 @@ def can_be_optimized(self) -> Result[tuple[bool, CodeOptimizationContext, dict[P
         # last time we could not find an optimization, maybe this time we do.
         # Random is before as a performance optimization, swapping the two 'and' statements has the same effect
         if random.random() > REPEAT_OPTIMIZATION_PROBABILITY and was_function_previously_optimized(  # noqa: S311
-            self.function_to_optimize, code_context
+            self.function_to_optimize, code_context, self.args
         ):
             return Failure("Function optimization previously attempted, skipping.")