from lm_eval.api.filter import Filter


class DecontaminationFilter(Filter):

    """
    A filter which evaluates
    """

    name = "track_decontamination"

    def __init__(self, path):
        """

        TODO: make sure only ever run one time on the train set (should this be cached as a class var? keyed by value for "path").
        should further cache result on a given (task_name, doc_id)
        """
        self._decontam_results = None

    def apply(self, reps):
        """
        Return {"no_contamination", "only_contamination"} keys for the 2 different subsets
        """
        pass
