Use airflow.models.dag.DAG in dev (apache#34615)

asapekia · Sep 26, 2023 · 2b57f39 · 2b57f39
1 parent 8f1418f
commit 2b57f39
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/dev/perf/dags/elastic_dag.py b/dev/perf/dags/elastic_dag.py
@@ -22,8 +22,8 @@
 from datetime import datetime, timedelta
 from enum import Enum
 
-from airflow import DAG
 from airflow.models.baseoperator import chain
+from airflow.models.dag import DAG
 from airflow.operators.bash import BashOperator
 
 # DAG File used in performance tests. Its shape can be configured by environment variables.

diff --git a/dev/perf/dags/perf_dag_1.py b/dev/perf/dags/perf_dag_1.py
@@ -22,7 +22,7 @@
 
 import datetime
 
-from airflow.models import DAG
+from airflow.models.dag import DAG
 from airflow.operators.bash_operator import BashOperator
 
 args = {

diff --git a/dev/perf/dags/perf_dag_2.py b/dev/perf/dags/perf_dag_2.py
@@ -22,7 +22,7 @@
 
 import datetime
 
-from airflow.models import DAG
+from airflow.models.dag import DAG
 from airflow.operators.bash import BashOperator
 
 args = {