Fix tests

apache · Mar 10, 2023 · 03fb0e4 · 03fb0e4
1 parent 26830cd
commit 03fb0e4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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},