from funcy import rcompose


class PredictionModelHandle:
    """Simplifies usage of ModelHandle instances for prediction purposes."""

    def __init__(self, model_handle):
        self.__predict = rcompose(model_handle.prep_images, model_handle.model.predict_proba)

    def predict(self, *args, **kwargs):
        return self.__predict(*args, **kwargs)

    def __call__(self, *args, **kwargs):
        return self.predict(*args, **kwargs)