from typing import Dict
import kserve
import logging
logging.basicConfig(level=kserve.constants.KSERVE_LOGLEVEL)

class MyBasicTransformer(kserve.Model):
    def __init__(self, name: str, predictor_host: str):
        super().__init__(name)
        self.predictor_host = predictor_host

    def preprocess(self, inputs: Dict) -> Dict:
        print(f"Transformer received inputs: {inputs}")
        return inputs

    def postprocess(self, inputs: Dict) -> Dict:
        return inputs