Allow arguments for py_tasks