# Path: datarobot_batch_scoring/consts.py
# class TargetType(object):
#     REGRESSION = 'Regression'
#     BINARY = 'Binary'
# 
# Path: datarobot_batch_scoring/exceptions.py
# class UnexpectedKeptColumnCount(Exception):
#     pass


import operator
import json
from datarobot_batch_scoring.consts import TargetType
from datarobot_batch_scoring.exceptions import UnexpectedKeptColumnCount



def format_data(result, batch, **opts):
    pred_name = opts.get('pred_name')
    keep_cols = opts.get('keep_cols')
    skip_row_id = opts.get('skip_row_id')
    fast_mode = opts.get('fast_mode')

    predictions = result['predictions']

# Path: datarobot_batch_scoring/consts.py
# class TargetType(object):
#     REGRESSION = 'Regression'
#     BINARY = 'Binary'
# 
# Path: datarobot_batch_scoring/exceptions.py
# class UnexpectedKeptColumnCount(Exception):
#     pass


# Path: datarobot_batch_scoring/api_response_handlers/api_v1.py
import operator
import json
from datarobot_batch_scoring.consts import TargetType
from datarobot_batch_scoring.exceptions import UnexpectedKeptColumnCount




def format_data(result, batch, **opts):
    pred_name = opts.get('pred_name')
    keep_cols = opts.get('keep_cols')
    skip_row_id = opts.get('skip_row_id')
    fast_mode = opts.get('fast_mode')

    predictions = result['predictions']

    if result['task'] == TargetType.BINARY: