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