This is an automated email from the ASF dual-hosted git repository. ash pushed a commit to branch task-sdk-first-code in repository https://gitbox.apache.org/repos/asf/airflow.git
commit 737e24544f4a0b3506c4a4cb94f26219892fe3de Author: Ash Berlin-Taylor <[email protected]> AuthorDate: Tue Oct 22 17:49:11 2024 +0100 [skip ci] --- airflow/serialization/serialized_objects.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/airflow/serialization/serialized_objects.py b/airflow/serialization/serialized_objects.py index e2e21a6686d..e595ef17449 100644 --- a/airflow/serialization/serialized_objects.py +++ b/airflow/serialization/serialized_objects.py @@ -22,6 +22,7 @@ import collections.abc import datetime import enum import inspect +import itertools import logging import weakref from functools import cache @@ -59,6 +60,7 @@ from airflow.models.taskinstancekey import TaskInstanceKey from airflow.models.tasklog import LogTemplate from airflow.models.xcom_arg import XComArg, deserialize_xcom_arg, serialize_xcom_arg from airflow.providers_manager import ProvidersManager +from airflow.sdk import BaseOperator as TaskSDKBaseOperator from airflow.serialization.dag_dependency import DagDependency from airflow.serialization.enums import DagAttributeTypes as DAT, Encoding from airflow.serialization.helpers import serialize_template_field @@ -1080,7 +1082,10 @@ class SerializedBaseOperator(BaseOperator, BaseSerialization): _CONSTRUCTOR_PARAMS = { k: v.default - for k, v in signature(BaseOperator.__init__).parameters.items() + for k, v in itertools.chain( + signature(BaseOperator.__init__).parameters.items(), + signature(TaskSDKBaseOperator.__init__).parameters.items(), + ) if v.default is not v.empty }
