def input_fn()

in src/inference_mxnet.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def input_fn(request_body, content_type):
    '''
    Deserialize the request body
    '''    
    if content_type == 'application/json':
        D = json.loads(request_body)

        short = D.get('short')
        image = b64decode(D['image'])
        x, _ = gcv.data.transforms.presets.yolo.transform_test(
            mx.image.imdecode(image), short=short
        )
        return x
    else:
        raise RuntimeError(f'Not support content-type: {content_type}')