fix tmp removal

Author: MashAliK

openevolve/controller.py

@@ -5,6 +5,7 @@
 import asyncio
 import logging
 import os
+import shutil
 import re
 import time
 import uuid
@@ -216,7 +217,7 @@ async def run(
         self.database.log_island_status()
 
         # create temp file to save database snapshots to for process workers to load from
-        temp_db_path = "temp/" + str(uuid.uuid4())
+        temp_db_path = "tmp/" + str(uuid.uuid4())
         self.database.save(temp_db_path, start_iteration)
 
         with concurrent.futures.ProcessPoolExecutor(
@@ -293,7 +294,7 @@ async def run(
                 except Exception:
                     logger.exception(f"Error in iteration {iteration}:")
                     continue
-        os.rmdir(temp_db_path)
+        shutil.rmtree(temp_db_path)
 
         # Get the best program using our tracking mechanism
         best_program = None

openevolve/database.py

@@ -10,7 +10,7 @@
 import time
 from dataclasses import asdict, dataclass, field
 from pathlib import Path
-from Levenshtein import distance, ratio
+from Levenshtein import ratio
 from filelock import FileLock, Timeout
 from typing import Any, Dict, List, Optional, Set, Tuple, Union
 
@@ -307,7 +307,8 @@ def save(self, path: Optional[str] = None, iteration: int = 0) -> None:
             logger.warning("No database path specified, skipping save")
             return
 
-        lock_path = save_path + ".lock"
+        lock_name = os.path.basename(save_path) + '.lock'
+        lock_path = os.path.join('tmp/locks', lock_name)
         try:
             with FileLock(lock_path, timeout=10):
                 # Create directory and remove old path if it exists
@@ -371,7 +372,8 @@ def load(self, path: str) -> None:
             logger.info(f"Loaded database metadata with last_iteration={self.last_iteration}")
 
         # Load programs
-        lock_path = path + ".lock"
+        lock_name = os.path.basename(path) + '.lock'
+        lock_path = os.path.join('tmp/locks', lock_name)
         programs_dir = os.path.join(path, "programs")
         try:
             with FileLock(lock_path, timeout=10):