Skip to content

Commit 71cd3bc

Browse files
committed
dont hardcode defaults
1 parent 0d2ddab commit 71cd3bc

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

codeflash/optimization/function_optimizer.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2733,8 +2733,7 @@ def establish_original_code_baseline(
27332733
testing_time=TOTAL_LOOPING_TIME_EFFECTIVE,
27342734
enable_coverage=False,
27352735
code_context=code_context,
2736-
pytest_min_loops=3 if subagent else 5,
2737-
pytest_max_loops=100 if subagent else 250,
2736+
**({"pytest_min_loops": 3, "pytest_max_loops": 100} if subagent else {}),
27382737
)
27392738
logger.debug(f"[BENCHMARK-DONE] Got {len(benchmarking_results.test_results)} benchmark results")
27402739
finally:
@@ -2937,8 +2936,7 @@ def run_optimized_candidate(
29372936
optimization_iteration=optimization_candidate_index,
29382937
testing_time=TOTAL_LOOPING_TIME_EFFECTIVE,
29392938
enable_coverage=False,
2940-
pytest_min_loops=3 if subagent else 5,
2941-
pytest_max_loops=100 if subagent else 250,
2939+
**({"pytest_min_loops": 3, "pytest_max_loops": 100} if subagent else {}),
29422940
)
29432941
finally:
29442942
if self.function_to_optimize.is_async:

0 commit comments

Comments
 (0)