From 0753ea3a2f3010a5550bd3ad88b95e26359bfddf Mon Sep 17 00:00:00 2001 From: Pankaj Singh <98807258+pankajastro@users.noreply.github.com> Date: Thu, 20 Jul 2023 11:39:20 +0530 Subject: [PATCH] Replace DummyOperator with emptyOperator (#1251) --- .circleci/integration-tests/master_dag.py | 4 ++-- .../providers/amazon/aws/example_dags/example_aws_nuke.py | 4 ++-- .../providers/core/example_dags/example_external_task.py | 4 ++-- .../providers/dbt/cloud/example_dags/example_dbt_cloud.py | 4 ++-- tests/core/triggers/test_external_task.py | 4 ++-- 5 files changed, 10 insertions(+), 10 deletions(-) diff --git a/.circleci/integration-tests/master_dag.py b/.circleci/integration-tests/master_dag.py index 9beca374e..af0465564 100644 --- a/.circleci/integration-tests/master_dag.py +++ b/.circleci/integration-tests/master_dag.py @@ -10,7 +10,7 @@ from airflow.models import DagRun from airflow.models.baseoperator import chain from airflow.operators.bash import BashOperator -from airflow.operators.dummy import DummyOperator +from airflow.operators.empty import EmptyOperator from airflow.operators.python import PythonOperator from airflow.operators.trigger_dagrun import TriggerDagRunOperator from airflow.utils.session import create_session @@ -357,7 +357,7 @@ def prepare_dag_dependency(task_info, execution_time): provide_context=True, ) - end = DummyOperator( + end = EmptyOperator( task_id="end", trigger_rule="all_success", ) diff --git a/astronomer/providers/amazon/aws/example_dags/example_aws_nuke.py b/astronomer/providers/amazon/aws/example_dags/example_aws_nuke.py index 74708fdf8..ec70aea64 100644 --- a/astronomer/providers/amazon/aws/example_dags/example_aws_nuke.py +++ b/astronomer/providers/amazon/aws/example_dags/example_aws_nuke.py @@ -6,7 +6,7 @@ from airflow import DAG from airflow.operators.bash import BashOperator -from airflow.operators.dummy import DummyOperator +from airflow.operators.empty import EmptyOperator from airflow.operators.python import PythonOperator from airflow.providers.slack.operators.slack_webhook import SlackWebhookOperator from airflow.utils.state import State @@ -104,7 +104,7 @@ def check_dag_status(**kwargs: Any) -> None: tags=["example", "aws-nuke"], is_paused_upon_creation=False, ) as dag: - start = DummyOperator(task_id="start") + start = EmptyOperator(task_id="start") get_airflow_version = BashOperator( task_id="get_airflow_version", bash_command="airflow version", do_xcom_push=True diff --git a/astronomer/providers/core/example_dags/example_external_task.py b/astronomer/providers/core/example_dags/example_external_task.py index 6d94751d8..8732ae899 100644 --- a/astronomer/providers/core/example_dags/example_external_task.py +++ b/astronomer/providers/core/example_dags/example_external_task.py @@ -4,7 +4,7 @@ from typing import Any from airflow import DAG -from airflow.operators.dummy import DummyOperator +from airflow.operators.empty import EmptyOperator from airflow.operators.python import PythonOperator from airflow.operators.trigger_dagrun import TriggerDagRunOperator from airflow.utils.state import State @@ -40,7 +40,7 @@ def check_dag_status(**kwargs: Any) -> None: default_args=default_args, tags=["example", "async", "core"], ) as dag: - start = DummyOperator(task_id="start") + start = EmptyOperator(task_id="start") # [START howto_sensor_external_task_async] waiting_for_task = ExternalTaskSensorAsync( diff --git a/astronomer/providers/dbt/cloud/example_dags/example_dbt_cloud.py b/astronomer/providers/dbt/cloud/example_dags/example_dbt_cloud.py index 7a2921222..7ce3d945f 100644 --- a/astronomer/providers/dbt/cloud/example_dags/example_dbt_cloud.py +++ b/astronomer/providers/dbt/cloud/example_dags/example_dbt_cloud.py @@ -5,7 +5,7 @@ from typing import Any from airflow import DAG -from airflow.operators.dummy import DummyOperator +from airflow.operators.empty import EmptyOperator from airflow.operators.python import PythonOperator from airflow.providers.dbt.cloud.operators.dbt import DbtCloudRunJobOperator from airflow.utils.state import State @@ -48,7 +48,7 @@ def check_dag_status(**kwargs: Any) -> None: tags=["example", "async", "dbt-cloud"], catchup=False, ) as dag: - start = DummyOperator(task_id="start") + start = EmptyOperator(task_id="start") # [START howto_operator_dbt_cloud_run_job_async] trigger_dbt_job_run_async = DbtCloudRunJobOperatorAsync( diff --git a/tests/core/triggers/test_external_task.py b/tests/core/triggers/test_external_task.py index f94b80283..f2171bf99 100644 --- a/tests/core/triggers/test_external_task.py +++ b/tests/core/triggers/test_external_task.py @@ -5,7 +5,7 @@ import asynctest import pytest from airflow import AirflowException -from airflow.operators.dummy import DummyOperator +from airflow.operators.empty import EmptyOperator from airflow.triggers.base import TriggerEvent from airflow.utils.state import DagRunState, TaskInstanceState @@ -35,7 +35,7 @@ async def test_task_state_trigger(self, session, dag): session.add(dag_run) session.commit() - external_task = DummyOperator(task_id=self.TASK_ID, dag=dag) + external_task = EmptyOperator(task_id=self.TASK_ID, dag=dag) instance = get_task_instance(external_task) session.add(instance) session.commit()