@@ -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]) ->
187187def 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