add docs for some functions, remove unused imports

Author: HenriqueAssumpcao

src/codeevolve/evolution.py

@@ -44,7 +44,6 @@ def select_parents(
     evolve_config: Dict[str, Any],
     gen_init_pop: bool,
     exploration: bool,
-    epoch: int,
     logger: logging.Logger,
 ) -> Tuple[Program, Program, List[Program]]:
     """
@@ -72,7 +71,6 @@ def select_parents(
             - migration_interval: Epochs between migrations (default: 20)
         gen_init_pop: Whether currently generating initial population
         exploration: Whether in exploration mode (vs exploitation)
-        epoch: Current epoch number
         logger: Logger for recording selection decisions
 
     Returns:
@@ -653,7 +651,6 @@ async def evolve_loop(
             evolve_config=evolve_config,
             gen_init_pop=gen_init_pop,
             exploration=exploration,
-            epoch=epoch,
             logger=logger,
         )
 

src/codeevolve/prompt/sampler.py

@@ -12,7 +12,6 @@
 
 from typing import Dict, List, Tuple, Optional
 from collections import deque
-import logging
 
 from codeevolve.lm import OpenAILM
 from codeevolve.database import Program, ProgramDatabase

src/codeevolve/scheduler.py

@@ -10,7 +10,7 @@
 #
 # ===--------------------------------------------------------------------------------------===#
 
-from typing import Any, Callable, Dict, List, Optional
+from typing import Dict, Optional
 from abc import ABC, abstractmethod
 
 import numpy as np
@@ -264,18 +264,45 @@ def reset(self, exploration_rate: Optional[float] = None) -> None:
 
 
 class CosineScheduler(ExplorationRateScheduler):
-    """ """
+    """Cosine annealing scheduler that oscillates exploration rate periodically.
+    
+    This scheduler uses a cosine wave to smoothly vary the exploration rate
+    between min_rate and max_rate over a fixed cycle length, allowing for
+    periodic transitions between exploration and exploitation.
+    
+    Attributes:
+        cycle_length: Number of epochs for one complete cosine cycle.
+    """
 
     def __init__(
         self, exploration_rate: float, max_rate: float, min_rate: float, cycle_length: int
     ):
+        """Initialize the cosine annealing scheduler.
+        
+        Args:
+            exploration_rate: Initial exploration rate.
+            max_rate: Maximum exploration rate bound.
+            min_rate: Minimum exploration rate bound.
+            cycle_length: Number of epochs per complete cosine cycle.
+            
+        Raises:
+            ValueError: If cycle_length is not positive.
+        """
         super().__init__(exploration_rate, max_rate, min_rate)
         if cycle_length <= 0:
             raise ValueError(f"cycle_length ({cycle_length}) must be positive")
         self.cycle_length = cycle_length
 
     def __call__(self, epoch: int, **kwargs) -> float:
-        """ """
+        """Compute exploration rate using cosine annealing.
+        
+        Args:
+            epoch: Current epoch number.
+            **kwargs: Additional arguments (ignored).
+            
+        Returns:
+            Updated exploration rate following cosine wave.
+        """
         cycle_progress: float = (epoch % self.cycle_length) / self.cycle_length
         cosine_factor: float = 0.5 * (1 + np.cos(np.pi * cycle_progress))
         rate: float = self.min_rate + (self.max_rate - self.min_rate) * cosine_factor

src/codeevolve/utils/logging_utils.py

@@ -22,10 +22,6 @@
 
 from codeevolve.islands import GlobalData
 
-from typing import Optional
-import logging
-
-
 class SizeLimitedFormatter(logging.Formatter):
     """Custom logging formatter that enforces a maximum message size.