Method __init__

src/benchmark/context.py:29–65

Initialize the benchmark context manager. Args: base_dir: Base directory for storing benchmark data save_path: Path to save benchmark configurations contract_path: Path to save benchmark contract

(self, 
                 base_dir: Optional[str] = None,
                 save_path: Optional[str] = None,
                 contract_path: Optional[str] = None,
                 **kwargs)

    contract_path: str = Field(default=None, description="The path to save the benchmark contract")
    
    def __init__(self, 
                 base_dir: Optional[str] = None,
                 save_path: Optional[str] = None,
                 contract_path: Optional[str] = None,
                 **kwargs):
        """Initialize the benchmark context manager.
        
        Args:
            base_dir: Base directory for storing benchmark data
            save_path: Path to save benchmark configurations
            contract_path: Path to save benchmark contract
        """
        super().__init__(**kwargs)
        
        if base_dir is not None:
            self.base_dir = assemble_project_path(base_dir)
        else:
            self.base_dir = assemble_project_path(os.path.join(config.workdir, "benchmark"))
        logger.info(f"| 📁 Benchmark context manager base directory: {self.base_dir}.")    
        os.makedirs(self.base_dir, exist_ok=True)
        
        if save_path is not None:
            self.save_path = assemble_project_path(save_path)
        else:
            self.save_path = os.path.join(self.base_dir, "benchmark.json")
        logger.info(f"| 📁 Benchmark context manager save path: {self.save_path}.")
        
        if contract_path is not None:
            self.contract_path = assemble_project_path(contract_path)
        else:
            self.contract_path = os.path.join(self.base_dir, "contract.md")
        logger.info(f"| 📁 Benchmark context manager contract path: {self.contract_path}.")

        self._benchmark_configs: Dict[str, BenchmarkConfig] = {}  # Current active configs (latest version)
        self._benchmark_history_versions: Dict[str, Dict[str, BenchmarkConfig]] = {}
        
        self._cleanup_registered = False
    
    async def initialize(self, benchmark_names: Optional[List[str]] = None):
        """Initialize the benchmark context manager."""