from image_prediction.estimator.adapter.adapter import EstimatorAdapter


class EstimatorPreprocessor:
    def __init__(self, estimator: EstimatorAdapter):
        self.estimator = estimator

    def predict(self, batch):
        return self.estimator.predict(batch)