[airflow] branch main updated: Use iterative loop to look for mapped parent (#34622)

[uranusjr]

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)