def output_fn()

in src/inference.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def output_fn(prediction, accept='application/jsonlines'):
    if accept == 'application/jsonlines':
        ret = ''
        for p in prediction:
            ret += json.dumps(p) + os.linesep
        return ret, accept
    raise Exception(f'Requested unsupported ContentType in Accept: {accept}')