Ensure that this works:

```python
@setup
def mytask():
    print("I am a setup task")

with dag_maker() as dag:
    mytask.override(task_id='newtask')

assert len(dag.task_group.children) == 1
setup_task = dag.task_group.children["newtask"]
assert setup_task._is_setup
```
and teardown also works
run_as_user: str | None = None,
    executor_config: dict | None = None,
    inlets: Any | None = None,
    outlets: Any | None = None,
    doc: str | None = None,
    doc_md: str | None = None,
    doc_json: str | None = None,
    doc_yaml: str | None = None,
    doc_rst: str | None = None,
    **kwargs,
) -> OperatorPartial:
    from airflow.models.dag import DagContext
    from airflow.utils.task_group import TaskGroupContext
    validate_mapping_kwargs(operator_class, "partial", kwargs)
    dag = dag or DagContext.get_current_dag()
    if dag:
        task_group = TaskGroupContext.get_current_task_group(dag)
    if task_group:
        task_id = task_group.child_id(task_id)
    
    partial_kwargs, partial_params = get_merged_defaults(
        dag=dag,
        task_group=task_group,
        task_params=params,
        task_default_args=kwargs.pop("default_args", None),
    )
    partial_kwargs.update(kwargs)
    
    partial_kwargs.setdefault("dag", dag)
    partial_kwargs.setdefault("task_group", task_group)