import abc


class Estimator(abc.ABC):
    def __init__(self, estimator):
        self.estimator = estimator

    @abc.abstractmethod
    def predict(self, batch):
        pass