def _user_module_transformer()

in src/sagemaker_mxnet_serving_container/handler_service.py [0:0]

• 18 lines of code
• 5 McCabe index (conditional complexity)

    def _user_module_transformer(model_dir=environment.model_dir):
        try:
            user_module = importlib.import_module(environment.Environment().module_name)
        except ModuleNotFoundError as e:
            logging.error("import_module exception: {}".format(e))
            raise ValueError('import_module exception: {}'.format(e))

        if hasattr(user_module, 'transform_fn'):
            return Transformer(default_inference_handler=DefaultMXNetInferenceHandler())

        model_fn = getattr(user_module, 'model_fn', DefaultMXNetInferenceHandler().default_model_fn)

        model = model_fn(model_dir)
        if isinstance(model, mx.module.BaseModule):
            return MXNetModuleTransformer()
        elif isinstance(model, mx.gluon.block.Block):
            return Transformer(default_inference_handler=DefaultGluonBlockInferenceHandler())
        else:
            raise ValueError('Unsupported model type: {}. Did you forget to implement '
                             '`transform_fn` or `model_fn` in your entry-point?'
                             .format(model.__class__.__name__))