(airflow) branch main updated: Pass more arguments to underlying DAG (#47898)

[uranusjr]

This is an automated email from the ASF dual-hosted git repository.

uranusjr pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new 6ff147ab9e4 Pass more arguments to underlying DAG (#47898)
6ff147ab9e4 is described below

commit 6ff147ab9e4e5abf8679c9a12569e4564b05f480
Author: Tzu-ping Chung <uranu...@gmail.com>
AuthorDate: Wed Mar 19 12:17:21 2025 +0800

    Pass more arguments to underlying DAG (#47898)
---
 task-sdk/src/airflow/sdk/definitions/asset/decorators.py     | 6 +++++-
 task-sdk/tests/task_sdk/definitions/test_asset_decorators.py | 6 ++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/task-sdk/src/airflow/sdk/definitions/asset/decorators.py 
b/task-sdk/src/airflow/sdk/definitions/asset/decorators.py
index ad9916e58a6..dea0cf1b95b 100644
--- a/task-sdk/src/airflow/sdk/definitions/asset/decorators.py
+++ b/task-sdk/src/airflow/sdk/definitions/asset/decorators.py
@@ -164,7 +164,8 @@ class _DAGFactory:
     on_failure_callback: None | DagStateChangeCallback | 
list[DagStateChangeCallback] = None
 
     access_control: dict[str, dict[str, Collection[str]]] | None = None
-    owner_links: dict[str, str] | None = None
+    owner_links: dict[str, str] = attrs.field(factory=dict)
+    tags: Collection[str] = attrs.field(factory=set)
 
     def create_dag(self, *, default_dag_id: str) -> DAG:
         from airflow.models.dag import DAG  # TODO: Use the SDK DAG when it 
works.
@@ -180,6 +181,9 @@ class _DAGFactory:
             params=self.params,
             on_success_callback=self.on_success_callback,
             on_failure_callback=self.on_failure_callback,
+            access_control=self.access_control,
+            owner_links=self.owner_links,
+            tags=self.tags,
             auto_register=True,
         )
 
diff --git a/task-sdk/tests/task_sdk/definitions/test_asset_decorators.py 
b/task-sdk/tests/task_sdk/definitions/test_asset_decorators.py
index 87923809024..afa94c0a7e3 100644
--- a/task-sdk/tests/task_sdk/definitions/test_asset_decorators.py
+++ b/task-sdk/tests/task_sdk/definitions/test_asset_decorators.py
@@ -176,6 +176,9 @@ class TestAssetDefinition:
             on_failure_callback=None,
             on_success_callback=None,
             params=None,
+            access_control=None,
+            owner_links={},
+            tags=set(),
             auto_register=True,
         )
         from_definition.assert_called_once_with(asset_definition)
@@ -200,6 +203,9 @@ class TestMultiAssetDefinition:
             on_failure_callback=None,
             on_success_callback=None,
             params=None,
+            access_control=None,
+            owner_links={},
+            tags=set(),
             auto_register=True,
         )
         from_definition.assert_called_once_with(definition)