def _replace_with_arrays(args, stmt, block, func_ir): new_args = [] for var in args: call_stmt = make_call(fix_df_array, [var], {}, block, func_ir, var.loc) insert_before(block, call_stmt, stmt) new_args.append(call_stmt.target) return new_args
def _replace_index_with_arrays(args, stmt, block, func_ir): new_args = [] call_stmt = make_call(fix_df_index, args, {}, block, func_ir, args[0].loc) insert_before(block, call_stmt, stmt) new_args.append(call_stmt.target) return new_args return new_args