diff --git a/airflow/providers/google/cloud/example_dags/example_cloud_sql_query.py b/airflow/providers/google/cloud/example_dags/example_cloud_sql_query.py index b94c8a0c40cce..040ce91515cfb 100644 --- a/airflow/providers/google/cloud/example_dags/example_cloud_sql_query.py +++ b/airflow/providers/google/cloud/example_dags/example_cloud_sql_query.py @@ -43,7 +43,7 @@ from pathlib import Path from urllib.parse import quote_plus -from airflow import models +from airflow.models.dag import DAG from airflow.providers.google.cloud.operators.cloud_sql import CloudSQLExecuteQueryOperator GCP_PROJECT_ID = os.environ.get("GCP_PROJECT_ID", "example-project") @@ -265,7 +265,7 @@ def get_absolute_path(path): tasks = [] -with models.DAG( +with DAG( dag_id="example_gcp_sql_query", start_date=datetime(2021, 1, 1), catchup=False, diff --git a/airflow/providers/google/cloud/example_dags/example_cloud_task.py b/airflow/providers/google/cloud/example_dags/example_cloud_task.py index 877eac15f3453..662199c23abc0 100644 --- a/airflow/providers/google/cloud/example_dags/example_cloud_task.py +++ b/airflow/providers/google/cloud/example_dags/example_cloud_task.py @@ -29,7 +29,7 @@ import os from datetime import datetime -from airflow import models +from airflow.models.dag import DAG from airflow.providers.google.cloud.sensors.tasks import TaskQueueEmptySensor GCP_PROJECT_ID = os.environ.get("GCP_PROJECT_ID", "example-project") @@ -37,7 +37,7 @@ QUEUE_NAME = os.environ.get("GCP_QUEUE_NAME", "testqueue") -with models.DAG( +with DAG( "example_gcp_cloud_tasks_sensor", start_date=datetime(2022, 8, 8), catchup=False, diff --git a/airflow/providers/google/cloud/example_dags/example_facebook_ads_to_gcs.py b/airflow/providers/google/cloud/example_dags/example_facebook_ads_to_gcs.py index 96da2ad126b90..8a94faf43ce59 100644 --- a/airflow/providers/google/cloud/example_dags/example_facebook_ads_to_gcs.py +++ b/airflow/providers/google/cloud/example_dags/example_facebook_ads_to_gcs.py @@ -25,8 +25,8 @@ from facebook_business.adobjects.adsinsights import AdsInsights -from airflow import models from airflow.models.baseoperator import chain +from airflow.models.dag import DAG from airflow.providers.google.cloud.operators.bigquery import ( BigQueryCreateEmptyDatasetOperator, BigQueryCreateEmptyTableOperator, @@ -57,7 +57,7 @@ PARAMETERS = {"level": "ad", "date_preset": "yesterday"} # [END howto_FB_ADS_variables] -with models.DAG( +with DAG( "example_facebook_ads_to_gcs", start_date=datetime(2021, 1, 1), catchup=False, diff --git a/airflow/providers/google/cloud/example_dags/example_looker.py b/airflow/providers/google/cloud/example_dags/example_looker.py index 36cc0f56280bd..1925d0fc2b78c 100644 --- a/airflow/providers/google/cloud/example_dags/example_looker.py +++ b/airflow/providers/google/cloud/example_dags/example_looker.py @@ -23,11 +23,11 @@ from datetime import datetime -from airflow import models +from airflow.models.dag import DAG from airflow.providers.google.cloud.operators.looker import LookerStartPdtBuildOperator from airflow.providers.google.cloud.sensors.looker import LookerCheckPdtBuildSensor -with models.DAG( +with DAG( dag_id="example_gcp_looker", start_date=datetime(2021, 1, 1), catchup=False, diff --git a/airflow/providers/google/cloud/example_dags/example_presto_to_gcs.py b/airflow/providers/google/cloud/example_dags/example_presto_to_gcs.py index 5c5ce985b5965..c89b85342c5b4 100644 --- a/airflow/providers/google/cloud/example_dags/example_presto_to_gcs.py +++ b/airflow/providers/google/cloud/example_dags/example_presto_to_gcs.py @@ -24,7 +24,7 @@ import re from datetime import datetime -from airflow import models +from airflow.models.dag import DAG from airflow.providers.google.cloud.operators.bigquery import ( BigQueryCreateEmptyDatasetOperator, BigQueryCreateExternalTableOperator, @@ -48,7 +48,7 @@ def safe_name(s: str) -> str: return re.sub("[^0-9a-zA-Z_]+", "_", s) -with models.DAG( +with DAG( dag_id="example_presto_to_gcs", start_date=datetime(2021, 1, 1), catchup=False, diff --git a/airflow/providers/google/cloud/example_dags/example_salesforce_to_gcs.py b/airflow/providers/google/cloud/example_dags/example_salesforce_to_gcs.py index c41e515b6547c..74a6b91cbe329 100644 --- a/airflow/providers/google/cloud/example_dags/example_salesforce_to_gcs.py +++ b/airflow/providers/google/cloud/example_dags/example_salesforce_to_gcs.py @@ -22,7 +22,7 @@ import os from datetime import datetime -from airflow import models +from airflow.models.dag import DAG from airflow.providers.google.cloud.operators.bigquery import ( BigQueryCreateEmptyDatasetOperator, BigQueryCreateEmptyTableOperator, @@ -43,7 +43,7 @@ SALESFORCE_CONN_ID = os.environ.get("SALESFORCE_CONN_ID", "salesforce_default") -with models.DAG( +with DAG( "example_salesforce_to_gcs", start_date=datetime(2021, 1, 1), catchup=False, diff --git a/airflow/providers/google/marketing_platform/example_dags/example_display_video.py b/airflow/providers/google/marketing_platform/example_dags/example_display_video.py index 9caf6eb509480..a54d7e575be11 100644 --- a/airflow/providers/google/marketing_platform/example_dags/example_display_video.py +++ b/airflow/providers/google/marketing_platform/example_dags/example_display_video.py @@ -24,7 +24,7 @@ from datetime import datetime from typing import cast -from airflow import models +from airflow.models.dag import DAG from airflow.models.xcom_arg import XComArg from airflow.providers.google.cloud.transfers.gcs_to_bigquery import GCSToBigQueryOperator from airflow.providers.google.marketing_platform.hooks.display_video import GoogleDisplayVideo360Hook @@ -87,7 +87,7 @@ START_DATE = datetime(2021, 1, 1) -with models.DAG( +with DAG( "example_display_video_misc", start_date=START_DATE, catchup=False, @@ -120,7 +120,7 @@ ) # [END howto_google_display_video_upload_line_items_operator] -with models.DAG( +with DAG( "example_display_video_sdf", start_date=START_DATE, catchup=False, @@ -168,7 +168,7 @@ # Task dependency created via `XComArgs`: # save_sdf_in_gcs >> upload_sdf_to_big_query -with models.DAG( +with DAG( "example_display_video_v2", start_date=START_DATE, catchup=False,