Merge branch 'main' of https://github.com/inter-co/science-codeevolve

Author: HenriqueAssumpcao

src/codeevolve/cli.py

@@ -172,9 +172,7 @@ def main() -> int:
 
     global_data: GlobalSyncData = GlobalSyncData(
         best_sol=global_best_sol,
-        early_stop_counter=mp.Value(
-            ctypes.c_int, early_stop_counter, lock=False
-        ),
+        early_stop_counter=mp.Value(ctypes.c_int, early_stop_counter, lock=False),
         early_stop_aux=mp.Value(ctypes.c_int, 0, lock=False),
         lock=mp.Lock(),
         barrier=mp.Barrier(parties=evolve_config["num_islands"]),

src/codeevolve/evaluator.py

@@ -222,9 +222,7 @@ def execute(
             eval_metrics: Dictionary of evaluation metrics if successful
         """
         effective_timeout: int = timeout_s if timeout_s is not None else self.timeout_s
-        self.logger.info(
-            f"Attempting to evaluate program (timeout={effective_timeout}s)..."
-        )
+        self.logger.info(f"Attempting to evaluate program (timeout={effective_timeout}s)...")
 
         extension: str = LANGUAGE_TO_EXTENSION.get(prog.language, DEFAULT_EXTENSION)
         returncode: int = 1

src/codeevolve/evolution.py

@@ -719,13 +719,22 @@ def _do_checkpoint(epoch_num: int) -> None:
         if isl_data.id == 0:
             elapsed_s: float = get_elapsed_time(global_data)
             cpus: int = global_data.cpu_count.value
-            save_run_metadata(args["out_dir"], epoch_num, elapsed_s, cpus, global_data.best_sol,global_data.early_stop_counter.value)
+            save_run_metadata(
+                args["out_dir"],
+                epoch_num,
+                elapsed_s,
+                cpus,
+                global_data.best_sol,
+                global_data.early_stop_counter.value,
+            )
             logger.info(f"Saved run metadata for epoch {epoch_num}.")
 
     meta_prompting: bool = evolve_config.get("meta_prompting", False)
     use_map_elites: bool = evolve_config.get("use_map_elites", False)
     exploration_rate: float = (
-        exploration_scheduler.value if exploration_scheduler is not None else evolve_config["exploration_rate"]
+        exploration_scheduler.value
+        if exploration_scheduler is not None
+        else evolve_config["exploration_rate"]
     )
     eval_budget: str = format_eval_budget(
         timeout_s=evaluator.timeout_s, max_mem_b=evaluator.max_mem_b
@@ -806,13 +815,13 @@ def _do_checkpoint(epoch_num: int) -> None:
 
         child_timeout: Optional[int] = None
         if not gen_init_pop and timeout_scheduler is not None:
-            child_timeout = int(timeout_scheduler(
-                epoch=epoch,
-                best_fitness=sol_db.programs[sol_db.best_prog_id].fitness,
-            ))
-            eval_budget = format_eval_budget(
-                timeout_s=child_timeout, max_mem_b=evaluator.max_mem_b
+            child_timeout = int(
+                timeout_scheduler(
+                    epoch=epoch,
+                    best_fitness=sol_db.programs[sol_db.best_prog_id].fitness,
+                )
             )
+            eval_budget = format_eval_budget(timeout_s=child_timeout, max_mem_b=evaluator.max_mem_b)
 
         child_sol, evolve_success = await generate_solution(
             ensemble=ensemble,
@@ -1356,9 +1365,15 @@ def setup_codeevolve_components(
     init_sol: Program
 
     if args["load_ckpt"]:
-        prompt_db, sol_db, evolve_state, init_prompt, init_sol, exploration_scheduler, timeout_scheduler = (
-            _initialize_from_checkpoint(args, exploration_scheduler, timeout_scheduler)
-        )
+        (
+            prompt_db,
+            sol_db,
+            evolve_state,
+            init_prompt,
+            init_sol,
+            exploration_scheduler,
+            timeout_scheduler,
+        ) = _initialize_from_checkpoint(args, exploration_scheduler, timeout_scheduler)
     else:
         prompt_db, sol_db, evolve_state, init_prompt, init_sol = _initialize_new_run(
             config, evolve_config, args, isl_data.id, evaluator, logger
@@ -1442,7 +1457,7 @@ async def codeevolve(
     components.logger.info("Waiting for other islands to finish setup...")
     global_data.barrier.wait()
     components.logger.info("All islands finished. Starting CodeEvolve loop.")
-    
+
     await codeevolve_loop(
         components.start_epoch,
         components.evolve_state,

src/codeevolve/runner.py

@@ -138,7 +138,9 @@ def async_run_codeevolve(
         try:
             isl_logs_dir: Path = run_args.get("logs_dir")
             if isl_logs_dir:
-                _write_island_crash_log(isl_logs_dir, error_msg, full_traceback, int(global_data.start_time.value))
+                _write_island_crash_log(
+                    isl_logs_dir, error_msg, full_traceback, int(global_data.start_time.value)
+                )
         except Exception:
             pass
 
@@ -247,7 +249,9 @@ def cleanup_log_daemon(
 # ---------------------------------------------------------------------------
 
 
-def _write_island_crash_log(isl_logs_dir: Path, error_msg: str, full_traceback: str, time: int = 0) -> None:
+def _write_island_crash_log(
+    isl_logs_dir: Path, error_msg: str, full_traceback: str, time: int = 0
+) -> None:
     """Writes detailed crash information to an island's crash log file.
 
     Called from within the child process where the exception occurred.
@@ -269,7 +273,9 @@ def _write_island_crash_log(isl_logs_dir: Path, error_msg: str, full_traceback:
         f.write(f"{'='*60}\n\n")
 
 
-def _write_crash_summary(out_dir: Path, island_id: int, exit_code: int, message: str, time: int = 0) -> None:
+def _write_crash_summary(
+    out_dir: Path, island_id: int, exit_code: int, message: str, time: int = 0
+) -> None:
     """Writes crash summary to the main output directory's crash log.
 
     Called from the parent process when it detects an island has crashed.
@@ -289,7 +295,9 @@ def _write_crash_summary(out_dir: Path, island_id: int, exit_code: int, message:
         f.write(f"Island: {island_id}\n")
         f.write(f"Exit Code: {exit_code}\n")
         f.write(f"Message: {message}\n")
-        f.write(f"See island_{island_id}/logs/{CRASH_LOG_FILE.format(time=time)} for full traceback.\n")
+        f.write(
+            f"See island_{island_id}/logs/{CRASH_LOG_FILE.format(time=time)} for full traceback.\n"
+        )
         f.write(f"{'='*60}\n\n")
 
 

src/codeevolve/scheduler.py

@@ -91,7 +91,6 @@ def reset(self, value: Optional[float] = None) -> None:
             self.value = value
 
 
-
 class ExponentialScheduler(Scheduler):
     """Exponential scheduler that scales a value geometrically over epochs.
 
@@ -107,9 +106,7 @@ class ExponentialScheduler(Scheduler):
         initial_value: Value at epoch 0 (used for reset).
     """
 
-    def __init__(
-        self, value: float, max_value: float, min_value: float, weight: float
-    ):
+    def __init__(self, value: float, max_value: float, min_value: float, weight: float):
         """
         Initialize the exponential scheduler.
 
@@ -276,9 +273,7 @@ class CosineScheduler(Scheduler):
         cycle_length: Number of epochs per complete cosine cycle.
     """
 
-    def __init__(
-        self, value: float, max_value: float, min_value: float, cycle_length: int
-    ):
+    def __init__(self, value: float, max_value: float, min_value: float, cycle_length: int):
         """Initialize the cosine annealing scheduler.
 
         Args:

src/codeevolve/utils/ckpt.py

@@ -129,7 +129,7 @@ def save_run_metadata(
     elapsed_time: float,
     cpu_count: int,
     global_best_sol: GlobalBestProg,
-    early_stop_counter: int
+    early_stop_counter: int,
 ) -> None:
     """Saves run metadata to a JSON file.
 
@@ -164,7 +164,7 @@ def save_run_metadata(
             "depth": global_best_sol.depth.value,
             "eval_metrics": dict(global_best_sol.eval_metrics),
         },
-        "early_stop_counter": early_stop_counter
+        "early_stop_counter": early_stop_counter,
     }
 
     with open(metadata_file, "w") as f:

tests/test_ckpt.py

@@ -210,7 +210,12 @@ def test_save_and_load_metadata(self, tmp_path: Path):
         best_sol.depth.value = 5
 
         save_run_metadata(
-            tmp_path, epoch=10, elapsed_time=120.5, cpu_count=8, global_best_sol=best_sol,early_stop_counter=5
+            tmp_path,
+            epoch=10,
+            elapsed_time=120.5,
+            cpu_count=8,
+            global_best_sol=best_sol,
+            early_stop_counter=5,
         )
 
         metadata: Optional[Dict[str, Any]] = load_run_metadata(tmp_path, epoch=10)
@@ -229,7 +234,12 @@ def test_load_missing_epoch(self, tmp_path: Path):
         """Tests loading metadata for a missing epoch returns empty dict."""
         best_sol: GlobalBestProg = GlobalBestProg()
         save_run_metadata(
-            tmp_path, epoch=10, elapsed_time=100.0, cpu_count=4, global_best_sol=best_sol, early_stop_counter=5
+            tmp_path,
+            epoch=10,
+            elapsed_time=100.0,
+            cpu_count=4,
+            global_best_sol=best_sol,
+            early_stop_counter=5,
         )
 
         metadata: Optional[Dict[str, Any]] = load_run_metadata(tmp_path, epoch=99)
@@ -239,10 +249,20 @@ def test_metadata_accumulates(self, tmp_path: Path):
         """Tests that multiple saves accumulate in the same file."""
         best_sol: GlobalBestProg = GlobalBestProg()
         save_run_metadata(
-            tmp_path, epoch=10, elapsed_time=100.0, cpu_count=4, global_best_sol=best_sol, early_stop_counter=5
+            tmp_path,
+            epoch=10,
+            elapsed_time=100.0,
+            cpu_count=4,
+            global_best_sol=best_sol,
+            early_stop_counter=5,
         )
         save_run_metadata(
-            tmp_path, epoch=20, elapsed_time=200.0, cpu_count=4, global_best_sol=best_sol, early_stop_counter=10
+            tmp_path,
+            epoch=20,
+            elapsed_time=200.0,
+            cpu_count=4,
+            global_best_sol=best_sol,
+            early_stop_counter=10,
         )
 
         metadata_file: Path = tmp_path / RUN_METADATA_FILE

tests/test_scheduler.py

@@ -39,13 +39,9 @@ def test_creation(self):
     def test_invalid_weight(self):
         """Tests that non-positive weight raises ValueError."""
         with pytest.raises(ValueError):
-            ExponentialScheduler(
-                value=0.5, max_value=1.0, min_value=0.01, weight=0.0
-            )
+            ExponentialScheduler(value=0.5, max_value=1.0, min_value=0.01, weight=0.0)
         with pytest.raises(ValueError):
-            ExponentialScheduler(
-                value=0.5, max_value=1.0, min_value=0.01, weight=-1.0
-            )
+            ExponentialScheduler(value=0.5, max_value=1.0, min_value=0.01, weight=-1.0)
 
     def test_decay_over_epochs(self):
         """Tests that value decays over epochs with weight < 1."""
@@ -261,16 +257,12 @@ class TestSchedulerBase:
     def test_invalid_min_max_value(self):
         """Tests that min_value > max_value raises ValueError."""
         with pytest.raises(ValueError):
-            ExponentialScheduler(
-                value=0.5, max_value=0.1, min_value=0.9, weight=0.99
-            )
+            ExponentialScheduler(value=0.5, max_value=0.1, min_value=0.9, weight=0.99)
 
     def test_value_outside_bounds(self):
         """Tests that value outside [min_value, max_value] raises ValueError."""
         with pytest.raises(ValueError):
-            ExponentialScheduler(
-                value=1.5, max_value=1.0, min_value=0.0, weight=0.99
-            )
+            ExponentialScheduler(value=1.5, max_value=1.0, min_value=0.0, weight=0.99)
 
     def test_reset_with_invalid_value(self):
         """Tests that reset with value outside bounds raises ValueError."""