Skip to content

Commit 639adec

Browse files
committed
Update end_to_end_test_utilities.py
1 parent f7d8d6a commit 639adec

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/scripts/end_to_end_test_utilities.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def build_command(
122122
) -> list[str]:
123123
python_path = "../../../codeflash/main.py" if "code_directories" in str(cwd) else "../codeflash/main.py"
124124

125-
base_command = ["python", python_path, "--file", config.file_path, "--no-pr"]
125+
base_command = ["uv","run", python_path, "--file", config.file_path, "--no-pr"]
126126

127127
if config.function_name:
128128
base_command.extend(["--function", config.function_name])
@@ -187,7 +187,7 @@ def validate_stdout_in_candidate(stdout: str, expected_in_stdout: list[str]) ->
187187
def run_trace_test(cwd: pathlib.Path, config: TestConfig, expected_improvement_pct: int) -> bool:
188188
test_root = cwd / "tests" / (config.test_framework or "")
189189
clear_directory(test_root)
190-
command = ["python", "-m", "codeflash.main", "optimize", "workload.py"]
190+
command = ["uv", "run", "-m", "codeflash.main", "optimize", "workload.py"]
191191
process = subprocess.Popen(
192192
command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True, cwd=str(cwd), env=os.environ.copy()
193193
)

0 commit comments

Comments
 (0)