From 91e149da997b3386427afbce511dae76cc131b3d Mon Sep 17 00:00:00 2001 From: Saurabh Misra Date: Sun, 6 Jul 2025 18:34:56 -0700 Subject: [PATCH] revert code under optimization when optimization found --- codeflash/optimization/function_optimizer.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index daa14f2b8..131d1e44e 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -1047,7 +1047,12 @@ def find_and_process_best_optimization( coverage_message=coverage_message, git_remote=self.args.git_remote, ) - if self.args.all or env_utils.get_pr_number() or (self.args.file and not self.args.function): + if ( + self.args.all + or env_utils.get_pr_number() + or self.args.replay_test + or (self.args.file and not self.args.function) + ): self.write_code_and_helpers( self.function_to_optimize_source_code, original_helper_code,