Source code for pina.loss.weighting_interface

"""Module for the Weighting Interface."""

from abc import ABCMeta, abstractmethod




class WeightingInterface(metaclass=ABCMeta):
    """
    Abstract base class for all loss weighting schemas. All weighting schemas
    should inherit from this class.
    """

    def __init__(self):
        """
        Initialization of the :class:`WeightingInterface` class.
        """
        self.condition_names = None



    @abstractmethod
    def aggregate(self, losses):
        """
        Aggregate the losses.

        :param dict losses: The dictionary of losses.
        """