17 lines
333 B
Python
17 lines
333 B
Python
from image_prediction.estimator.estimator import Estimator
|
|
|
|
|
|
class DummyEstimator:
|
|
@staticmethod
|
|
def predict(_):
|
|
return True
|
|
|
|
|
|
class EstimatorMock(Estimator):
|
|
|
|
def __init__(self, estimator):
|
|
super().__init__(estimator=estimator)
|
|
|
|
def predict(self, batch):
|
|
return self.estimator.predict(batch)
|