|
76 | 76 | from codeflash.lsp.lsp_message import LspCodeMessage, LspMarkdownMessage, LSPMessageId |
77 | 77 | from codeflash.models.ExperimentMetadata import ExperimentMetadata |
78 | 78 | from codeflash.models.models import ( |
79 | | - AIServiceCodeRepairRequest, |
80 | 79 | BestOptimization, |
81 | 80 | CandidateEvaluationContext, |
82 | 81 | CodeOptimizationContext, |
@@ -763,7 +762,6 @@ def process_single_candidate( |
763 | 762 | file_path_to_helper_classes: dict[Path, set[str]], |
764 | 763 | eval_ctx: CandidateEvaluationContext, |
765 | 764 | all_refinements_data: list[AIServiceRefinerRequest], |
766 | | - ai_service_client: AiServiceClient, |
767 | 765 | exp_type: str, |
768 | 766 | function_references: str, |
769 | 767 | ) -> BestOptimization | None: |
@@ -883,9 +881,7 @@ def process_single_candidate( |
883 | 881 | optimized_line_profiler_results=best_optimization.line_profiler_test_results["str_out"], |
884 | 882 | function_references=function_references, |
885 | 883 | ) |
886 | | - original_line_profiler_results=original_code_baseline.line_profile_results["str_out"], |
887 | | - optimized_line_profiler_results=best_optimization.line_profiler_test_results["str_out"], |
888 | | - |
| 884 | + ) |
889 | 885 | # Display runtime information |
890 | 886 | if is_LSP_enabled(): |
891 | 887 | lsp_log(LspMarkdownMessage(markdown=tree_to_markdown(tree))) |
@@ -939,7 +935,12 @@ def determine_best_candidate( |
939 | 935 | ) |
940 | 936 |
|
941 | 937 | processor = CandidateProcessor( |
942 | | - candidates, future_line_profile_results, all_refinements_data, self.future_all_code_repair, self.aiservice_client, self.executor |
| 938 | + candidates, |
| 939 | + future_line_profile_results, |
| 940 | + all_refinements_data, |
| 941 | + self.future_all_code_repair, |
| 942 | + self.aiservice_client, |
| 943 | + self.executor, |
943 | 944 | ) |
944 | 945 | candidate_index = 0 |
945 | 946 |
|
|
0 commit comments