Fix DB utils for Airflow Backwards compatability tests - import not e…

…xisting (apache#42522)
astronomer · Sep 26, 2024 · 3749dbd · 3749dbd
1 parent 84e8cdf
commit 3749dbd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_utils/db.py b/tests/test_utils/db.py
@@ -35,7 +35,6 @@
     Variable,
     XCom,
 )
-from airflow.models.backfill import Backfill, BackfillDagRun
 from airflow.models.dag import DagOwnerAttributes
 from airflow.models.dagcode import DagCode
 from airflow.models.dagwarning import DagWarning
@@ -68,6 +67,8 @@ def clear_db_runs():
 
 
 def clear_db_backfills():
+    from airflow.models.backfill import Backfill, BackfillDagRun
+
     with create_session() as session:
         session.query(BackfillDagRun).delete()
         session.query(Backfill).delete()