def combine_train_infer()

in src/utils.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

def combine_train_infer(train_file, infer_dir):
    """Combine training and inference datasets as one data frame"""

    train_df = pd.read_feather(train_file)

    time_range = range(len([f for f in os.listdir(infer_dir) if 'feather' in f]))
    infer_df_list = [pd.read_feather(f'{infer_dir}/{t}.feather') for t in time_range]

    comb_df_list = []
    train_df.index = [-1] * len(train_df)

    comb_df_list.append(train_df)

    for t in time_range:
        df = infer_df_list[t]
        df.index = [t] * len(df)

        comb_df_list.append(df)

    return pd.concat(comb_df_list), train_df, infer_df_list