Airflow - chain

 

from __future__ import annotations

import pendulum

from airflow.providers.standard.operators.empty import EmptyOperator
from airflow.sdk import DAG, chain


with DAG(
    dag_id='ZZH_DAG_1',
    schedule=None,
    start_date=pendulum.datetime(2021, 1, 1, tz='UTC'),
    catchup=False
) as dag:
    start = EmptyOperator(task_id='start')

    [task_1, task_2, task_3, task_4, task_5, task_6] = [EmptyOperator(task_id=f'task_{i}') for i in range(1, 7)]

    end = EmptyOperator(task_id='end')

    chain(start, [task_1, task_2], [task_3, task_4], [task_5, task_6], end)