def input_fn()

in inference/sklearn/inference.py [0:0]

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

def input_fn(input_data, content_type):

    if content_type == "text/csv":
        df = pd.read_csv(StringIO(input_data), header=None, index_col=False, sep=",")

        first_row = df.iloc[0:1].values[0].tolist()

        if len(df.columns) == len(feature_columns_names):
            print("column length is correct")

            if set(first_row) == set(feature_columns_names):
                print("the row contains header, remove the row")
                df = df.iloc[1:]
                df.reset_index(drop=True, inplace=True)

            df.columns = feature_columns_names

        return df
    else:
        raise ValueError("{} not supported by script!".format(content_type))