Source code for optimization.optimizer_interface

import abc
from typing import Callable, Dict
from optuna import Trial



class OptunaOptimizer(abc.ABC):



    @abc.abstractmethod
    def optimize(self, objective_function: Callable[[Trial], float]) -> Dict:
        r"""
        Optimize a model given objective function using Optuna.

        Args:
            objective_function (Callable[[Trial], float]): The objective function to optimize.

        Returns:
            Dict: The best parameters found by the optimizer.
        """
        pass