From bed2fa1ef334bd2070e39153a183d615a1f63851 Mon Sep 17 00:00:00 2001 From: Ephraim Anierobi Date: Wed, 15 Jan 2025 16:57:18 +0100 Subject: [PATCH] Fix compact tests --- tests_common/pytest_plugin.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests_common/pytest_plugin.py b/tests_common/pytest_plugin.py index 48701bcaf0e41..dab785c5b77e4 100644 --- a/tests_common/pytest_plugin.py +++ b/tests_common/pytest_plugin.py @@ -832,7 +832,10 @@ def __exit__(self, type, value, traceback): return dag.clear(session=self.session) - dag.bulk_write_to_db(self.bundle_name, None, [dag], session=self.session) + if AIRFLOW_V_3_0_PLUS: + dag.bulk_write_to_db(self.bundle_name, None, [dag], session=self.session) + else: + dag.sync_to_db(session=self.session) if dag.access_control: from airflow.www.security_appless import ApplessAirflowSecurityManager