From 2fab7a93a24982c12136d53d3de931c2d2dcb53d Mon Sep 17 00:00:00 2001 From: Kaxil Naik Date: Fri, 29 Nov 2024 16:48:03 +0000 Subject: [PATCH] Revert "Fix flakey test in DAG serialization (#44480)" This reverts commit cd5ccf0abf5027cf457b36a8a2bee397dcb538b4. --- task_sdk/src/airflow/sdk/definitions/dag.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/task_sdk/src/airflow/sdk/definitions/dag.py b/task_sdk/src/airflow/sdk/definitions/dag.py index d02480c8ab1bc..20621ba148fbd 100644 --- a/task_sdk/src/airflow/sdk/definitions/dag.py +++ b/task_sdk/src/airflow/sdk/definitions/dag.py @@ -991,7 +991,7 @@ def get_serialized_fields(cls): "fail_stop", "schedule", } - cls.__serialized_fields = frozenset(a.name for a in attrs.fields(cls)) - exclusion_list + cls.__serialized_fields = frozenset(vars(DAG(dag_id="test", schedule=None))) - exclusion_list return cls.__serialized_fields def get_edge_info(self, upstream_task_id: str, downstream_task_id: str) -> EdgeInfoType: