pierrejeambrun commented on code in PR #48771:
URL: https://github.com/apache/airflow/pull/48771#discussion_r2027793291


##########
airflow-core/src/airflow/api_fastapi/core_api/services/ui/grid.py:
##########
@@ -115,9 +117,11 @@ def _fill_task_group_map(
             task_nodes[task_node.task_id] = {
                 "is_group": False,
                 "parent_id": parent_node.node_id if parent_node else None,
-                "task_count": task_nodes[parent_node.node_id]["task_count"]
-                if _is_task_node_mapped_task_group(parent_node) and parent_node
-                else [1],
+                "task_count": (
+                    task_nodes[parent_node.node_id]["task_count"]
+                    if _is_task_node_mapped_task_group(parent_node) and 
parent_node
+                    else [1]
+                ),

Review Comment:
   ```suggestion
                   "task_count": task_nodes[parent_node.node_id]["task_count"]
                   if _is_task_node_mapped_task_group(parent_node) and 
parent_node
                   else [1],
   ```



##########
airflow-core/src/airflow/api_fastapi/core_api/services/ui/grid.py:
##########
@@ -237,9 +255,9 @@ def fill_task_instance_summaries(
                 queued_dttm=ti_queued_dttm,
                 child_states=child_states,
                 task_count=_get_total_task_count(run_id, 
task_node_map[task_id]["task_count"], session),
-                state=TaskInstanceState[overall_ti_state.upper()]
-                if overall_ti_state != "no_status"
-                else None,
+                state=(
+                    TaskInstanceState[overall_ti_state.upper()] if 
overall_ti_state != "no_status" else None
+                ),

Review Comment:
   ```suggestion
                   state=TaskInstanceState[overall_ti_state.upper()]
                   if overall_ti_state != "no_status"
                   else None,
   ```



##########
airflow-core/src/airflow/api_fastapi/core_api/services/ui/grid.py:
##########
@@ -137,12 +141,14 @@ def _get_total_task_count(
     run_id: str, task_count: list[int | MappedTaskGroup | MappedOperator], 
session: SessionDep
 ) -> int:
     return sum(
-        node
-        if isinstance(node, int)
-        else (
-            DBBaseOperator.get_mapped_ti_count(node, run_id=run_id, 
session=session) or 0
-            if isinstance(node, (MappedTaskGroup, MappedOperator))
-            else node
+        (
+            node
+            if isinstance(node, int)
+            else (
+                DBBaseOperator.get_mapped_ti_count(node, run_id=run_id, 
session=session) or 0
+                if isinstance(node, (MappedTaskGroup, MappedOperator))
+                else node
+            )

Review Comment:
   ```suggestion
           node
           if isinstance(node, int)
           else (
               DBBaseOperator.get_mapped_ti_count(node, run_id=run_id, 
session=session) or 0
               if isinstance(node, (MappedTaskGroup, MappedOperator))
               else node
   ```



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to