pass --async to E2E

Author: KRRT7

tests/scripts/end_to_end_test_async.py

@@ -9,6 +9,7 @@ def run_test(expected_improvement_pct: int) -> bool:
         file_path="main.py",
         expected_unit_tests=0,
         min_improvement_x=0.1,
+        enable_async=True,
         coverage_expectations=[
             CoverageExpectation(
                 function_name="retry_with_backoff",

tests/scripts/end_to_end_test_utilities.py

@@ -27,6 +27,7 @@ class TestConfig:
     trace_mode: bool = False
     coverage_expectations: list[CoverageExpectation] = field(default_factory=list)
     benchmarks_root: Optional[pathlib.Path] = None
+    enable_async: bool = False
 
 
 def clear_directory(directory_path: str | pathlib.Path) -> None:
@@ -132,6 +133,8 @@ def build_command(
         )
     if benchmarks_root:
         base_command.extend(["--benchmark", "--benchmarks-root", str(benchmarks_root)])
+    if config.enable_async:
+        base_command.append("--async")
     return base_command