|
8 | 8 | from codeflash.cli_cmds.console import logger |
9 | 9 | from codeflash.code_utils.code_utils import custom_addopts, get_run_tmp_file |
10 | 10 | from codeflash.code_utils.compat import IS_POSIX, SAFE_SYS_EXECUTABLE |
11 | | -from codeflash.code_utils.config_consts import get_total_looping_time |
| 11 | +from codeflash.code_utils.config_consts import TOTAL_LOOPING_TIME_EFFECTIVE |
12 | 12 | from codeflash.code_utils.coverage_utils import prepare_coverage_files |
13 | 13 | from codeflash.models.models import TestFiles, TestType |
14 | 14 |
|
@@ -37,7 +37,7 @@ def run_behavioral_tests( |
37 | 37 | pytest_timeout: int | None = None, |
38 | 38 | pytest_cmd: str = "pytest", |
39 | 39 | verbose: bool = False, |
40 | | - pytest_target_runtime_seconds: int = get_total_looping_time(), |
| 40 | + pytest_target_runtime_seconds: int = TOTAL_LOOPING_TIME_EFFECTIVE, |
41 | 41 | enable_coverage: bool = False, |
42 | 42 | ) -> tuple[Path, subprocess.CompletedProcess, Path | None, Path | None]: |
43 | 43 | if test_framework == "pytest": |
@@ -66,7 +66,7 @@ def run_behavioral_tests( |
66 | 66 | "--codeflash_loops_scope=session", |
67 | 67 | "--codeflash_min_loops=1", |
68 | 68 | "--codeflash_max_loops=1", |
69 | | - f"--codeflash_seconds={pytest_target_runtime_seconds}", # TODO : This is unnecessary, update the plugin to not ask for this |
| 69 | + f"--codeflash_seconds={pytest_target_runtime_seconds}", |
70 | 70 | ] |
71 | 71 |
|
72 | 72 | result_file_path = get_run_tmp_file(Path("pytest_results.xml")) |
@@ -151,7 +151,7 @@ def run_line_profile_tests( |
151 | 151 | cwd: Path, |
152 | 152 | test_framework: str, |
153 | 153 | *, |
154 | | - pytest_target_runtime_seconds: float = get_total_looping_time(), |
| 154 | + pytest_target_runtime_seconds: float = TOTAL_LOOPING_TIME_EFFECTIVE, |
155 | 155 | verbose: bool = False, |
156 | 156 | pytest_timeout: int | None = None, |
157 | 157 | pytest_min_loops: int = 5, # noqa: ARG001 |
@@ -237,7 +237,7 @@ def run_benchmarking_tests( |
237 | 237 | cwd: Path, |
238 | 238 | test_framework: str, |
239 | 239 | *, |
240 | | - pytest_target_runtime_seconds: float = get_total_looping_time(), |
| 240 | + pytest_target_runtime_seconds: float = TOTAL_LOOPING_TIME_EFFECTIVE, |
241 | 241 | verbose: bool = False, |
242 | 242 | pytest_timeout: int | None = None, |
243 | 243 | pytest_min_loops: int = 5, |
|
0 commit comments