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 cbe28c3a8b2426c6c4198c206c9b6192f566fbc2 Author: Ash Berlin-Taylor <[email protected]> AuthorDate: Fri Oct 25 14:30:40 2024 +0100 Fix default pool --- task_sdk/src/airflow/sdk/definitions/baseoperator.py | 2 +- task_sdk/src/airflow/sdk/definitions/dag.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/task_sdk/src/airflow/sdk/definitions/baseoperator.py b/task_sdk/src/airflow/sdk/definitions/baseoperator.py index 5796a6bbc1f..7b78aaca83d 100644 --- a/task_sdk/src/airflow/sdk/definitions/baseoperator.py +++ b/task_sdk/src/airflow/sdk/definitions/baseoperator.py @@ -767,7 +767,7 @@ class BaseOperator(AbstractOperator, metaclass=BaseOperatorMeta): self.retries = retries self.queue = queue # TODO: Task-SDK: pull this default name from Pool constant? - self.pool = "default" if pool is None else pool + self.pool = "default_pool" if pool is None else pool self.pool_slots = pool_slots if self.pool_slots < 1: dag_str = f" in dag {dag.dag_id}" if dag else "" diff --git a/task_sdk/src/airflow/sdk/definitions/dag.py b/task_sdk/src/airflow/sdk/definitions/dag.py index a8f222fd8ad..2bb15e9f2df 100644 --- a/task_sdk/src/airflow/sdk/definitions/dag.py +++ b/task_sdk/src/airflow/sdk/definitions/dag.py @@ -841,8 +841,7 @@ class DAG: t.upstream_task_ids.intersection_update(dag.task_dict) t.downstream_task_ids.intersection_update(dag.task_dict) - if len(dag.tasks) < len(self.tasks): - dag.partial = True + dag.partial = len(dag.tasks) < len(self.tasks) return dag
