from image_prediction.model.model import Model class ModelMock(Model): def __init__(self, estimator): super().__init__(estimator=estimator)