Skip to content

Commit 6764261

Browse files
committed
fix: Also use DAG in taskflow test in Airflow 3.0
1 parent 91aad8c commit 6764261

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/dags/test_dbt_dags.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -313,10 +313,8 @@ def test_dbt_operators_in_taskflow_dag(
313313
profiles_file,
314314
):
315315
"""Assert DAG contains correct dbt operators when running."""
316-
if AIRFLOW_V_3_1_PLUS:
316+
if AIRFLOW_V_3_0_PLUS:
317317
dag = taskflow_dag
318-
elif AIRFLOW_V_3_0_PLUS:
319-
dag = DAG.from_sdk_dag(taskflow_dag)
320318
else:
321319
dag = taskflow_dag
322320

0 commit comments

Comments
 (0)