From 25180d876a062c904ee49fed56faf4fe9e6c9fe1 Mon Sep 17 00:00:00 2001 From: Amogh Desai Date: Wed, 11 Jun 2025 13:49:53 +0530 Subject: [PATCH] Rename in GetXComCount to --- task-sdk/src/airflow/sdk/execution_time/supervisor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/task-sdk/src/airflow/sdk/execution_time/supervisor.py b/task-sdk/src/airflow/sdk/execution_time/supervisor.py index 7e6b89043d221..3fb5a9877a9a1 100644 --- a/task-sdk/src/airflow/sdk/execution_time/supervisor.py +++ b/task-sdk/src/airflow/sdk/execution_time/supervisor.py @@ -1107,8 +1107,8 @@ def _handle_request(self, msg: ToSupervisor, log: FilteringBoundLogger): xcom_result = XComResult.from_xcom_response(xcom) resp = xcom_result elif isinstance(msg, GetXComCount): - len = self.client.xcoms.head(msg.dag_id, msg.run_id, msg.task_id, msg.key) - resp = XComCountResponse(len=len) + xcom_count = self.client.xcoms.head(msg.dag_id, msg.run_id, msg.task_id, msg.key) + resp = XComCountResponse(len=xcom_count) elif isinstance(msg, GetXComSequenceItem): xcom = self.client.xcoms.get_sequence_item( msg.dag_id, msg.run_id, msg.task_id, msg.key, msg.offset