def update_step(func_name, args, step_name, caller_line):
    if states.workflow.dag_mode_enabled():
        step_name = _update_dag_tasks(func_name, states._dag_caller_line, states._upstream_dag_task, states._upstream_dag_depends_logic, args, step_name=step_name)
        states._upstream_dag_task = [step_name]
    elif states._run_concurrent_lock:
        step_name = _update_steps('concurrent_func_name', states._concurrent_func_line, args, func_name)
    else:
        step_name = _update_steps(func_name, caller_line, args)
    return step_name
def update_step(func_name, args, step_name, caller_line):
    if states.workflow.dag_mode_enabled():
        step_name = _update_dag_tasks(func_name, states._dag_caller_line, states._upstream_dag_task, states._upstream_dag_depends_logic, args, step_name=step_name)
        states._upstream_dag_task = [step_name]
    elif states._run_concurrent_lock:
        step_name = _update_steps('concurrent_func_name', states._concurrent_func_line, args, func_name)
    else:
        step_name = _update_steps(func_name, caller_line, args)
    return step_name