lower threshold and cleanup comments

Author: KRRT7

.github/workflows/e2e-futurehouse-structure.yaml

@@ -24,7 +24,7 @@ jobs:
       COLUMNS: 110
       MAX_RETRIES: 3
       RETRY_DELAY: 5
-      EXPECTED_IMPROVEMENT_PCT: 10
+      EXPECTED_IMPROVEMENT_PCT: 5
       CODEFLASH_END_TO_END: 1
     steps:
       - name: 🛎️ Checkout

codeflash/optimization/function_optimizer.py

@@ -1615,7 +1615,7 @@ def establish_original_code_baseline(
                     test_files=self.test_files,
                     optimization_iteration=0,
                     testing_time=total_looping_time,
-                    enable_coverage=True,  # Always enable coverage with pytest
+                    enable_coverage=True,
                     code_context=code_context,
                 )
             finally:

tests/scripts/end_to_end_test_futurehouse.py

@@ -8,7 +8,7 @@ def run_test(expected_improvement_pct: int) -> bool:
     config = TestConfig(
         file_path="src/aviary/common_tags.py",
         expected_unit_tests=1,
-        min_improvement_x=0.1,
+        min_improvement_x=0.05,
         coverage_expectations=[
             CoverageExpectation(
                 function_name="find_common_tags",

tests/scripts/end_to_end_test_utilities.py

@@ -85,7 +85,6 @@ def run_codeflash_command(
 
     path_to_file = cwd / config.file_path
     file_contents = path_to_file.read_text("utf-8")
-    # Check if tests/pytest exists, otherwise use tests/ directly
     pytest_dir = cwd / "tests" / "pytest"
     test_root = pytest_dir if pytest_dir.is_dir() else cwd / "tests"
 
@@ -130,7 +129,6 @@ def build_command(
 
     if config.function_name:
         base_command.extend(["--function", config.function_name])
-    # Always use pytest - test framework is specified via tests-root and module-root
     base_command.extend(["--tests-root", str(test_root), "--module-root", str(cwd)])
     if benchmarks_root:
         base_command.extend(["--benchmark", "--benchmarks-root", str(benchmarks_root)])
@@ -189,7 +187,6 @@ def validate_stdout_in_candidate(stdout: str, expected_in_stdout: list[str]) ->
 
 
 def run_trace_test(cwd: pathlib.Path, config: TestConfig, expected_improvement_pct: int) -> bool:
-    # Check if tests/pytest exists, otherwise use tests/ directly
     pytest_dir = cwd / "tests" / "pytest"
     test_root = pytest_dir if pytest_dir.is_dir() else cwd / "tests"
     clear_directory(test_root)