Skip to content

Commit a4a2bb5

Browse files
committed
refactor: Replace task chaining with chain function for improved readability in DAG
1 parent 1291947 commit a4a2bb5

1 file changed

Lines changed: 9 additions & 11 deletions

File tree

dags/tpu_observability/tpu_info_format_validation_dags.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -724,16 +724,14 @@ def generate_second_node_pool_name(
724724
setups=create_node_pool,
725725
)
726726

727-
# Airflow uses >> for task chaining, which is pointless for pylint.
728-
# pylint: disable=pointless-statement
729-
(
730-
cluster_info
731-
>> create_node_pool
732-
>> apply_time
733-
>> pod_names
734-
>> wait_for_job_start
735-
>> verification_group
736-
>> cleanup_workload
737-
>> cleanup_node_pool
727+
chain(
728+
cluster_info,
729+
create_node_pool,
730+
apply_time,
731+
pod_names,
732+
wait_for_job_start,
733+
verification_group,
734+
cleanup_workload,
735+
cleanup_node_pool,
738736
)
739737
# pylint: enable=pointless-statement

0 commit comments

Comments
 (0)