def output_fn()

in 08_projects/modelbuild/pipelines/endtoendmlsm/deploy/sklearn/inference.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

def output_fn(prediction, accept):
    if accept == "application/json":
        instances = []
        for row in prediction.tolist():
            instances.append({"features": row})
        json_output = {"instances": instances}

        return worker.Response(json.dumps(json_output), mimetype=accept)
    elif accept == 'text/csv':
        return worker.Response(encoders.encode(prediction, accept), mimetype=accept)
    else:
        raise RuntimeException("{} accept type is not supported.".format(accept))