def train_model(context, df): context.log.info(struct_to_string(df)) model = train(df) return model
def parse_df(context, df): context.log.info(struct_to_string(df)) result_df = df[:5] return result_df