import numpy as np

from image_prediction.estimator.estimator import Estimator


class KerasEstimator(Estimator):
    def __init__(self, estimator):
        super().__init__(estimator)

    def predict(self, batch: np.array):
        self.estimator.predict(batch)