linter

Author: codelion

openevolve/cli.py

@@ -44,31 +44,19 @@ def parse_args() -> argparse.Namespace:
         choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
         default="INFO",
     )
-    
+
     parser.add_argument(
         "--checkpoint",
         help="Path to checkpoint directory to resume from (e.g., openevolve_output/checkpoints/checkpoint_50)",
-        default=None
-    )
-    
-    parser.add_argument(
-        "--api-base",
-        help="Base URL for the LLM API",
-        default=None
-    )
-    
-    parser.add_argument(
-        "--primary-model",
-        help="Primary LLM model name",
-        default=None
+        default=None,
     )
-    
-    parser.add_argument(
-        "--secondary-model",
-        help="Secondary LLM model name",
-        default=None
-    )
-    
+
+    parser.add_argument("--api-base", help="Base URL for the LLM API", default=None)
+
+    parser.add_argument("--primary-model", help="Primary LLM model name", default=None)
+
+    parser.add_argument("--secondary-model", help="Secondary LLM model name", default=None)
+
     return parser.parse_args()
 
 
@@ -118,16 +106,18 @@ async def main_async() -> int:
             config_path=args.config if config is None else None,
             output_dir=args.output,
         )
-        
+
         # Load from checkpoint if specified
         if args.checkpoint:
             if not os.path.exists(args.checkpoint):
                 print(f"Error: Checkpoint directory '{args.checkpoint}' not found")
                 return 1
             print(f"Loading checkpoint from {args.checkpoint}")
             openevolve.database.load(args.checkpoint)
-            print(f"Checkpoint loaded successfully (iteration {openevolve.database.last_iteration})")
-        
+            print(
+                f"Checkpoint loaded successfully (iteration {openevolve.database.last_iteration})"
+            )
+
         # Override log level if specified
         if args.log_level:
             logging.getLogger().setLevel(getattr(logging, args.log_level))
@@ -137,25 +127,30 @@ async def main_async() -> int:
             iterations=args.iterations,
             target_score=args.target_score,
         )
-        
+
         # Get the checkpoint path
         checkpoint_dir = os.path.join(openevolve.output_dir, "checkpoints")
         latest_checkpoint = None
         if os.path.exists(checkpoint_dir):
-            checkpoints = [os.path.join(checkpoint_dir, d) for d in os.listdir(checkpoint_dir) 
-                          if os.path.isdir(os.path.join(checkpoint_dir, d))]
+            checkpoints = [
+                os.path.join(checkpoint_dir, d)
+                for d in os.listdir(checkpoint_dir)
+                if os.path.isdir(os.path.join(checkpoint_dir, d))
+            ]
             if checkpoints:
-                latest_checkpoint = sorted(checkpoints, key=lambda x: int(x.split("_")[-1]) if "_" in x else 0)[-1]
-        
+                latest_checkpoint = sorted(
+                    checkpoints, key=lambda x: int(x.split("_")[-1]) if "_" in x else 0
+                )[-1]
+
         print(f"\nEvolution complete!")
         print(f"Best program metrics:")
         for name, value in best_program.metrics.items():
             print(f"  {name}: {value:.4f}")
-            
+
         if latest_checkpoint:
             print(f"\nLatest checkpoint saved at: {latest_checkpoint}")
             print(f"To resume, use: --checkpoint {latest_checkpoint}")
-        
+
         return 0
 
     except Exception as e:

openevolve/controller.py

@@ -156,9 +156,11 @@ async def run(
         # Main evolution loop
         start_iteration = self.database.last_iteration
         total_iterations = start_iteration + max_iterations
-        
-        logger.info(f"Starting evolution from iteration {start_iteration} for {max_iterations} iterations (total: {total_iterations})")
-        
+
+        logger.info(
+            f"Starting evolution from iteration {start_iteration} for {max_iterations} iterations (total: {total_iterations})"
+        )
+
         for i in range(start_iteration, total_iterations):
             iteration_start = time.time()
 
@@ -354,7 +356,7 @@ def _save_checkpoint(self, iteration: int) -> None:
         # Save the database
         checkpoint_path = os.path.join(checkpoint_dir, f"checkpoint_{iteration}")
         self.database.save(checkpoint_path, iteration)
-        
+
         logger.info(f"Saved checkpoint at iteration {iteration} to {checkpoint_path}")
 
     def _save_best_program(self, program: Optional[Program] = None) -> None:

openevolve/database.py

@@ -80,10 +80,10 @@ def __init__(self, config: DatabaseConfig):
 
         # Track the absolute best program separately
         self.best_program_id: Optional[str] = None
-        
+
         # Track the last iteration number (for resuming)
         self.last_iteration: int = 0
-        
+
         # Load database from disk if path is provided
         if config.db_path and os.path.exists(config.db_path):
             self.load(config.db_path)
@@ -255,7 +255,7 @@ def get_top_programs(self, n: int = 10, metric: Optional[str] = None) -> List[Pr
             )
 
         return sorted_programs[:n]
-    
+
     def save(self, path: Optional[str] = None, iteration: int = 0) -> None:
         """
         Save the database to disk
@@ -312,9 +312,9 @@ def load(self, path: str) -> None:
             self.archive = set(metadata.get("archive", []))
             self.best_program_id = metadata.get("best_program_id")
             self.last_iteration = metadata.get("last_iteration", 0)
-            
+
             logger.info(f"Loaded database metadata with last_iteration={self.last_iteration}")
-        
+
         # Load programs
         programs_dir = os.path.join(path, "programs")
         if os.path.exists(programs_dir):