Skip to content

Commit 443cb4d

Browse files
authored
Merge pull request #747 from codeflash-ai/fix_looping_issue
import variable correctly
2 parents cdd16bb + 4c8b76f commit 443cb4d

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

codeflash/verification/test_runner.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from codeflash.cli_cmds.console import logger
99
from codeflash.code_utils.code_utils import custom_addopts, get_run_tmp_file
1010
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
1212
from codeflash.code_utils.coverage_utils import prepare_coverage_files
1313
from codeflash.models.models import TestFiles, TestType
1414

@@ -37,7 +37,7 @@ def run_behavioral_tests(
3737
pytest_timeout: int | None = None,
3838
pytest_cmd: str = "pytest",
3939
verbose: bool = False,
40-
pytest_target_runtime_seconds: int = get_total_looping_time(),
40+
pytest_target_runtime_seconds: int = TOTAL_LOOPING_TIME_EFFECTIVE,
4141
enable_coverage: bool = False,
4242
) -> tuple[Path, subprocess.CompletedProcess, Path | None, Path | None]:
4343
if test_framework == "pytest":
@@ -66,7 +66,7 @@ def run_behavioral_tests(
6666
"--codeflash_loops_scope=session",
6767
"--codeflash_min_loops=1",
6868
"--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}",
7070
]
7171

7272
result_file_path = get_run_tmp_file(Path("pytest_results.xml"))
@@ -151,7 +151,7 @@ def run_line_profile_tests(
151151
cwd: Path,
152152
test_framework: str,
153153
*,
154-
pytest_target_runtime_seconds: float = get_total_looping_time(),
154+
pytest_target_runtime_seconds: float = TOTAL_LOOPING_TIME_EFFECTIVE,
155155
verbose: bool = False,
156156
pytest_timeout: int | None = None,
157157
pytest_min_loops: int = 5, # noqa: ARG001
@@ -237,7 +237,7 @@ def run_benchmarking_tests(
237237
cwd: Path,
238238
test_framework: str,
239239
*,
240-
pytest_target_runtime_seconds: float = get_total_looping_time(),
240+
pytest_target_runtime_seconds: float = TOTAL_LOOPING_TIME_EFFECTIVE,
241241
verbose: bool = False,
242242
pytest_timeout: int | None = None,
243243
pytest_min_loops: int = 5,

0 commit comments

Comments
 (0)