repair the code after refinement if needed

Author: mohammedahmed18

codeflash/optimization/function_optimizer.py

@@ -221,7 +221,12 @@ def _process_code_repair(self) -> OptimizedCandidate | None:
 
     def is_done(self) -> bool:
         """Check if processing is complete."""
-        return self.line_profiler_done and self.refinement_done and self.candidate_queue.empty()
+        return (
+            self.line_profiler_done
+            and self.refinement_done
+            and len(self.future_all_code_repair) == 0
+            and self.candidate_queue.empty()
+        )
 
 
 class FunctionOptimizer: