def args(task):
    """Get the arguments for the current task."""

    if istask(task):
        return task[1:]
    if isinstance(task, list):
        return task
    return ()
Get the arguments for the current task.
src/flash/core/serve/dag/rewrite.py