Merge pull request codeflash-ai#520 from codeflash-ai/revert-replay-test-source-files

revert code under optimization when optimization found [CF-685]

Author: misrasaurabh1

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,