Source code for moosefs.merging_strategies.base_merger

from ..core.feature import Feature




class MergingStrategy:
    """Abstract base for merging strategies.

    Strategies can be "set-based" or "rank-based" depending on how they merge
    the per-selector outputs.
    """



    def __init__(self, strategy_type: str) -> None:
        """Initialize the strategy.

        Args:
            strategy_type: Either "set-based" or "rank-based".
        """
        self.strategy_type = strategy_type




    def merge(self, data: list, num_features_to_select: int, **kwargs) -> list:
        """Merge input data according to the strategy.

        Subclasses must implement this method.

        Args:
            data: List of Feature lists (one list per selector) or a single list.
            num_features_to_select: Number of top features to return.
            **kwargs: Strategy-specific options.

        Returns:
            A list of merged features (or names depending on strategy).

        Raises:
            NotImplementedError: If not implemented in a subclass.
        """
        raise NotImplementedError("Subclasses must implement this method")




    def is_set_based(self) -> bool:
        """Return True if the strategy is set-based."""
        return self.strategy_type == "set-based"




    def is_rank_based(self) -> bool:
        """Return True if the strategy is rank-based."""
        return self.strategy_type == "rank-based"




    def _validate_input(self, subsets: list) -> None:
        """Validate that ``subsets`` contains Feature objects.

        Args:
            subsets: A list of Feature or a list of Feature lists.

        Raises:
            ValueError: If empty or containing invalid types.
        """
        if not subsets:
            raise ValueError("Subsets cannot be empty.")

        if isinstance(subsets[0], list):  # List of lists case
            if not all(isinstance(sub, list) and sub for sub in subsets):
                raise ValueError("Subsets cannot contain empty lists.")
            if not all(
                isinstance(feature, Feature) for sub in subsets for feature in sub
            ):
                raise ValueError("Subsets must contain Feature objects.")
        else:  # Single list case
            if not all(isinstance(feature, Feature) for feature in subsets):
                raise ValueError("Subsets must contain Feature objects.")