potiuk commented on code in PR #37297:
URL: https://github.com/apache/airflow/pull/37297#discussion_r1485043421
##########
airflow/models/taskinstance.py:
##########
@@ -450,7 +450,16 @@ def _execute_callable(context, **execute_callable_kwargs):
else:
xcom_value = None
if xcom_value is not None: # If the task returns a result, push an
XCom containing it.
- task_instance.xcom_push(key=XCOM_RETURN_KEY, value=xcom_value,
session=session)
+ if task_to_execute.multiple_outputs:
+ if not isinstance(xcom_value, Mapping):
+ raise AirflowException(
+ "The returned value from the task is not a Mapping.
The `multiple_outputs`"
+ "parameter is set to True, so the result has to be a
Mapping."
+ )
+ for key, value in xcom_value.items():
Review Comment:
We do not limit the size in single output, so I think we should not do it in
multiple output case either.
--
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: commits-unsubscr...@airflow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
