File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -205,7 +205,8 @@ def basic_dag(
205205
206206 dbt_seed >> dbt_run >> dbt_test
207207
208- sync_dag_to_db (dag )
208+ if AIRFLOW_V_3_1_PLUS :
209+ sync_dag_to_db (dag )
209210 return dag
210211
211212
@@ -315,7 +316,10 @@ def prepare_dbt_project_dir() -> str:
315316 dbt_seed >> dbt_run >> dbt_test
316317
317318 d = generate_dag ()
318- sync_dag_to_db (d )
319+
320+ if AIRFLOW_V_3_1_PLUS :
321+ sync_dag_to_db (d )
322+
319323 return d
320324
321325
@@ -457,7 +461,8 @@ def target_connection_dag(
457461
458462 dbt_seed >> dbt_run >> dbt_test
459463
460- sync_dag_to_db (dag )
464+ if AIRFLOW_V_3_1_PLUS :
465+ sync_dag_to_db (dag )
461466 return dag
462467
463468
@@ -536,7 +541,9 @@ def test_example_basic_dag(
536541 dbt_run .target = "test"
537542 dbt_run .profile = "default"
538543
539- sync_dag_to_db (dag )
544+ if AIRFLOW_V_3_1_PLUS :
545+ sync_dag_to_db (dag )
546+
540547 dagrun = _create_dagrun (
541548 dag ,
542549 state = DagRunState .RUNNING ,
@@ -587,7 +594,9 @@ def test_example_dbt_project_in_github_dag(
587594 assert dag is not None
588595 assert len (dag .tasks ) == 3
589596
590- sync_dag_to_db (dag )
597+ if AIRFLOW_V_3_1_PLUS :
598+ sync_dag_to_db (dag )
599+
591600 dagrun = _create_dagrun (
592601 dag ,
593602 state = DagRunState .RUNNING ,
@@ -639,7 +648,9 @@ def test_example_complete_dbt_workflow_dag(
639648 assert dag is not None
640649 assert len (dag .tasks ) == 5
641650
642- sync_dag_to_db (dag )
651+ if AIRFLOW_V_3_1_PLUS :
652+ sync_dag_to_db (dag )
653+
643654 dagrun = _create_dagrun (
644655 dag ,
645656 state = DagRunState .RUNNING ,
You can’t perform that action at this time.
0 commit comments