Skip to content

Commit

Permalink
Use iterative loop to look for mapped parent (#34622)
Browse files Browse the repository at this point in the history
(cherry picked from commit d9ba152)
  • Loading branch information
uranusjr authored and ephraimbuddy committed Oct 5, 2023
1 parent 784d162 commit eb9274e
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions airflow/www/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,7 @@
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
Expand Down Expand Up @@ -425,14 +425,9 @@ def set_overall_state(record):
**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)
Expand Down

0 comments on commit eb9274e

Please sign in to comment.