def input_fn()

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

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

def input_fn(input_data, content_type):
    print(input_data)
    
    if content_type == 'text/csv':
        df = pd.read_csv(StringIO(input_data), header=None)
        if len(df.columns) == len(feature_columns_names):
            df.columns = feature_columns_names 
        return df
    else:
        raise ValueError("{} not supported by script!".format(content_type))