fixup! Update airflow/models/dag.py

apache · Feb 28, 2022 · ec3fea6 · ec3fea6
1 parent 7ce123b
commit ec3fea6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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)}