def _replace_call(stmt, new_call, args, block, func_ir):
func = stmt.value
data_args = args['data']
columns_args = args['columns']
index_args = args.get('index')
if index_args is None: # index arg was omitted
none_stmt = declare_constant(None, block, func_ir, stmt.loc)
index_args = none_stmt.target
index_args = [index_args]
all_args = data_args + index_args + columns_args
call = Expr.call(new_call, all_args, {}, func.loc)
stmt.value = call
def _replace_call(stmt, new_call, args, block, func_ir):
func = stmt.value
data_args = args['data']
columns_args = args['columns']
index_args = args.get('index')
if index_args is None:
none_stmt = declare_constant(None, block, func_ir, stmt.loc)
index_args = [none_stmt.target]
else:
index_args = RewriteDataFrame._replace_with_arrays([index_args], stmt, block, func_ir)
data_args = RewriteDataFrame._replace_with_arrays(data_args, stmt, block, func_ir)
all_args = data_args + index_args + columns_args
call = Expr.call(new_call, all_args, {}, func.loc)
stmt.value = call
def _replace_call(stmt, new_call, args, block, func_ir):
func = stmt.value
data_args = args['data']
columns_args = args['columns']
index_args = args.get('index')
data_args = RewriteDataFrame._replace_data_with_arrays(
data_args, stmt, block, func_ir)
if index_args is None: # index arg was omitted
none_stmt = declare_constant(None, block, func_ir, stmt.loc)
index_args = none_stmt.target
index_and_data_args = [index_args] + data_args
index_args = RewriteDataFrame._replace_index_with_arrays(
index_and_data_args, stmt, block, func_ir)
all_args = data_args + index_args + columns_args
call = Expr.call(new_call, all_args, {}, func.loc)
stmt.value = call