fix tests.

Author: codelion

openevolve/controller.py

@@ -9,7 +9,7 @@
 import time
 import uuid
 from pathlib import Path
-from typing import List, Optional, Tuple, Union
+from typing import Any, Dict, List, Optional, Tuple, Union
 import traceback
 
 from openevolve.config import Config, load_config

openevolve/database.py

@@ -499,7 +499,7 @@ def _distribute_programs_to_islands(self) -> None:
 
         logger.info(f"Distributed {len(program_ids)} programs across {len(self.islands)} islands")
 
-    def _save_program(self, program: Program, base_path: Optional[str] = None) -> None:
+    def _save_program(self, program: Program, base_path: Optional[str] = None, prompts: Optional[Dict[str, Dict[str, str]]] = None) -> None:
         """
         Save a program to disk