def output_fn()

in src/similarity/batch.py [0:0]

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

def output_fn(prediction_output, accept=OUTPUT_CONTENT_TYPE):
    logger.info('Serializing the generated output for '+accept)
    if accept == OUTPUT_CONTENT_TYPE:
        stream = StringIO()
        for i in range(len(prediction_output)):
            stream.write(image_names[i]+','+str(prediction_output[i])+'\n')
        return stream.getvalue()
    raise Exception('Requested unsupported ContentType in Accept: ' + accept)