This is an automated email from the ASF dual-hosted git repository.
uranusjr pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new d9ba152c15 Use iterative loop to look for mapped parent (#34622)
d9ba152c15 is described below
commit d9ba152c15dd50baa1fef41a63424225ba8ddd47
Author: Tzu-ping Chung <uranu...@gmail.com>
AuthorDate: Wed Sep 27 08:43:25 2023 +0800
Use iterative loop to look for mapped parent (#34622)
---
airflow/www/views.py | 9 ++-------
1 file changed, 2 insertions(+), 7 deletions(-)
diff --git a/airflow/www/views.py b/airflow/www/views.py
index ba6d4045c5..6fe8a32ea9 100644
--- a/airflow/www/views.py
+++ b/airflow/www/views.py
@@ -122,7 +122,7 @@ from airflow.utils.net import get_hostname
from airflow.utils.session import NEW_SESSION, create_session, provide_session
from airflow.utils.state import DagRunState, State, TaskInstanceState
from airflow.utils.strings import to_boolean
-from airflow.utils.task_group import MappedTaskGroup, TaskGroup,
task_group_to_dict
+from airflow.utils.task_group import TaskGroup, task_group_to_dict
from airflow.utils.timezone import td_format, utcnow
from airflow.version import version
from airflow.www import auth, utils as wwwutils
@@ -426,14 +426,9 @@ def dag_to_grid(dag: DagModel, dag_runs: Sequence[DagRun],
session: Session):
**setup_teardown_type,
}
- def check_group_is_mapped(tg: TaskGroup | None) -> bool:
- if tg is None:
- return False
- return isinstance(tg, MappedTaskGroup) or
check_group_is_mapped(tg.parent_group)
-
# Task Group
task_group = item
- group_is_mapped = check_group_is_mapped(task_group)
+ group_is_mapped = next(task_group.iter_mapped_task_groups(), None) is
not None
children = [
task_group_to_grid(child, grouped_tis,
is_parent_mapped=group_is_mapped)
