10 lines
267 B
Python
10 lines
267 B
Python
from image_prediction.estimator.adapter.adapter import EstimatorAdapter
|
|
|
|
|
|
class EstimatorPreprocessor:
|
|
def __init__(self, estimator: EstimatorAdapter):
|
|
self.estimator = estimator
|
|
|
|
def predict(self, batch):
|
|
return self.estimator.predict(batch)
|