Update end_to_end_test_utilities.py

KRRT7 · KRRT7 · commit d51a7417dec0 · 2025-07-01T19:25:34.000-07:00
diff --git a/tests/scripts/end_to_end_test_utilities.py b/tests/scripts/end_to_end_test_utilities.py
@@ -187,7 +187,7 @@ 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:
     test_root = cwd / "tests" / (config.test_framework or "")
     clear_directory(test_root)
-    command = ["python", "-m", "codeflash", "optimize", "workload.py"]
+    command = ["python", "-m", "codeflash.main", "optimize", "workload.py"]
     process = subprocess.Popen(
         command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, cwd=str(cwd), env=os.environ.copy()
     )

Original file line number	Diff line number	Diff line change
`@@ -187,7 +187,7 @@ def validate_stdout_in_candidate(stdout: str, expected_in_stdout: list[str]) ->`
`187`	`187`	`def run_trace_test(cwd: pathlib.Path, config: TestConfig, expected_improvement_pct: int) -> bool:`
`188`	`188`	`test_root = cwd / "tests" / (config.test_framework or "")`
`189`	`189`	`clear_directory(test_root)`
`190`		`- command = ["python", "-m", "codeflash", "optimize", "workload.py"]`
	`190`	`+ command = ["python", "-m", "codeflash.main", "optimize", "workload.py"]`
`191`	`191`	`process = subprocess.Popen(`
`192`	`192`	`command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, cwd=str(cwd), env=os.environ.copy()`
`193`	`193`	`)`