diff --git a/cosmos/core/airflow.py b/cosmos/core/airflow.py index 191dfcd6a..4eaeb2df5 100644 --- a/cosmos/core/airflow.py +++ b/cosmos/core/airflow.py @@ -10,7 +10,7 @@ logger = logging.getLogger(__name__) -def get_airflow_task(task: Task, dag: DAG, task_group: 'TaskGroup | None' = None) -> BaseOperator: +def get_airflow_task(task: Task, dag: DAG, task_group: "TaskGroup | None" = None) -> BaseOperator: """ Get the Airflow Operator class for a Task.