airflow-14563- pushing review comments.

apache · Aug 14, 2022 · 139cce4 · 139cce4
1 parent ab02251
commit 139cce4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airflow/models/dag.py b/airflow/models/dag.py
@@ -2220,10 +2220,10 @@ def filter_task_group(group, parent_group):
     def has_task(self, task_id: str):
         return task_id in self.task_dict
 
-    def has_task_group(self, task_group_id: str):
+    def has_task_group(self, task_group_id: str) -> bool:
         return task_group_id in self.task_group_dict
 
-    @property
+    @cached_property
     def task_group_dict(self):
         return {k: v for k, v in self._task_group.get_task_group_dict().items() if k is not None}