Skip to content

Commit eac2ff9

Browse files
kaxilLefteris Gilmaz
authored and
Lefteris Gilmaz
committed
Fix XCom value deserialization in Task SDK (apache#45254)
Fixing this so main works with most of the XCom example. But proper fix will be part of apache#45231
1 parent e5ddef0 commit eac2ff9

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

task_sdk/src/airflow/sdk/execution_time/task_runner.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -195,13 +195,12 @@ def xcom_pull(
195195
if TYPE_CHECKING:
196196
assert isinstance(msg, XComResult)
197197

198-
value = msg.value
199-
if value is not None:
198+
if msg.value is not None:
200199
from airflow.models.xcom import XCom
201200

202201
# TODO: Move XCom serialization & deserialization to Task SDK
203202
# https://github.com/apache/airflow/issues/45231
204-
return XCom.deserialize_value(value)
203+
return XCom.deserialize_value(msg) # type: ignore[arg-type]
205204
return default
206205

207206
def xcom_push(self, key: str, value: Any):

0 commit comments

Comments
 (0)