def create_inputs_map(values_map, input_config):
inputs = {}
for input_name, input_config_item in input_config.items():
if util.is_str(input_config_item):
inputs[input_name] = values_map[input_config_item]
elif util.is_int(input_config_item):
inputs[input_name] = values_map[input_config_item]
elif util.is_list(input_config_item):
inputs[input_name] = [values_map[f] for f in input_config_item]
else:
raise CortexException("invalid column inputs")
return inputs
def cast_compound_type(value, type_str):
allowed_types = type_str.split("|")
if consts.VALUE_TYPE_INT in allowed_types:
if util.is_int(value):
return value
if consts.VALUE_TYPE_FLOAT in allowed_types:
if util.is_int(value):
return float(value)
if util.is_float(value):
return value
if consts.VALUE_TYPE_STRING in allowed_types:
if util.is_str(value):
return value
if consts.VALUE_TYPE_BOOL in allowed_types:
if util.is_bool(value):
return value
raise UserException(
"unsupported input type (expected type {}, got {})".format(
util.data_type_str(type_str), util.user_obj_str(value)
)
)
def create_inputs_from_features_map(features_values_map, feature_input_config):
inputs = {}
for input_name, input_config_item in feature_input_config.items():
if util.is_str(input_config_item):
inputs[input_name] = features_values_map[input_config_item]
elif util.is_int(input_config_item):
inputs[input_name] = features_values_map[input_config_item]
elif util.is_list(input_config_item):
inputs[input_name] = [
features_values_map[f] for f in input_config_item
]
else:
raise CortexException("invalid feature inputs")
return inputs