Skip to content

Commit ac2a3a1

Browse files
Merge branch 'main' of github.com:codeflash-ai/codeflash into limit-refined-candidates
2 parents e895c52 + 890243c commit ac2a3a1

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

codeflash/optimization/function_optimizer.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -575,9 +575,11 @@ def handle_successful_candidate(
575575
576576
Returns the BestOptimization and optional benchmark tree.
577577
"""
578-
line_profile_test_results = self.line_profiler_step(
579-
code_context=code_context, original_helper_code=original_helper_code, candidate_index=candidate_index
580-
)
578+
with progress_bar("Running line-by-line profiling"):
579+
line_profile_test_results = self.line_profiler_step(
580+
code_context=code_context, original_helper_code=original_helper_code, candidate_index=candidate_index
581+
)
582+
581583
eval_ctx.record_line_profiler_result(candidate.optimization_id, line_profile_test_results["str_out"])
582584

583585
replay_perf_gain = {}

0 commit comments

Comments
 (0)