from image_prediction.service_estimator.service_estimator import ServiceEstimator class ServiceEstimatorMock(ServiceEstimator): def __init__(self, estimator, classes): super().__init__(estimator=estimator, classes=classes)