from image_prediction.estimator.adapter.adapter import EstimatorAdapter class Preprocessor: def __init__(self): pass # def preprocess(self, batch): # return (map(image_to_normalized_tensor, batch)) def __call__(self, batch): return self.preprocess(batch)