From 03fb0e41814b6e91b3cb9bdb5d00bb3fcc946ed6 Mon Sep 17 00:00:00 2001
From: Jed Cunningham <jedcunningham@apache.org>
Date: Fri, 10 Mar 2023 16:29:47 -0700
Subject: [PATCH] Fix tests

---
 tests/serialization/test_dag_serialization.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/serialization/test_dag_serialization.py b/tests/serialization/test_dag_serialization.py
index 68b72f05f8077..6e620b2aca7df 100644
--- a/tests/serialization/test_dag_serialization.py
+++ b/tests/serialization/test_dag_serialization.py
@@ -159,6 +159,8 @@ def detect_task_dependencies(task: Operator) -> DagDependency | None:  # type: i
                 "_task_type": "BashOperator",
                 "_task_module": "airflow.operators.bash",
                 "pool": "default_pool",
+                "_is_setup": False,
+                "_is_teardown": False,
                 "executor_config": {
                     "__type": "dict",
                     "__var": {
@@ -188,6 +190,8 @@ def detect_task_dependencies(task: Operator) -> DagDependency | None:  # type: i
                 "_operator_name": "@custom",
                 "_task_module": "tests.test_utils.mock_operators",
                 "pool": "default_pool",
+                "_is_setup": False,
+                "_is_teardown": False,
             },
         ],
         "schedule_interval": {"__type": "timedelta", "__var": 86400.0},