From ec3fea6e322f25465139b7f7527167bc275dc312 Mon Sep 17 00:00:00 2001 From: Ephraim Anierobi Date: Mon, 28 Feb 2022 18:35:26 +0100 Subject: [PATCH] fixup! Update airflow/models/dag.py --- airflow/models/dag.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/airflow/models/dag.py b/airflow/models/dag.py index e52c4c2875d16..b36d7fab5e11c 100644 --- a/airflow/models/dag.py +++ b/airflow/models/dag.py @@ -2631,8 +2631,12 @@ def validate_schedule_and_params(self): ) @provide_session - def validate_task_pools(self, session=NEW_SESSION): - """Validates and raise exception if any task in a dag is using a non-existent pool""" + def validate_task_pools(self, session: Session = NEW_SESSION): + """ + Validates and raise exception if any task in a dag is using a non-existent pool + + :meta private: + """ from airflow.models.pool import Pool pools = {p.pool for p in Pool.get_pools(session)}