diff --git a/airflow/executors/workloads.py b/airflow/executors/workloads.py index 00ca97405a04c..f42a0fe062ff7 100644 --- a/airflow/executors/workloads.py +++ b/airflow/executors/workloads.py @@ -61,7 +61,7 @@ class TaskInstance(BaseModel): pool_slots: int queue: str priority_weight: int - executor_config: dict | None = None + executor_config: dict | None = Field(default=None, exclude=True) # TODO: Task-SDK: Can we replace TastInstanceKey with just the uuid across the codebase? @property diff --git a/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py b/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py index 098adde9c1a99..cbe3eec2309b4 100644 --- a/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py +++ b/providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py @@ -264,7 +264,7 @@ def send_task_to_executor( if AIRFLOW_V_3_0_PLUS: if TYPE_CHECKING: assert isinstance(args, workloads.BaseWorkload) - args = (args.model_dump_json(exclude={"ti": {"executor_config"}}),) + args = (args.model_dump_json(),) try: with timeout(seconds=OPERATION_TIMEOUT): result = task_to_run.apply_async(args=args, queue=queue)