From e288df6fe2d0a068e7fe51da701955c25b603dc1 Mon Sep 17 00:00:00 2001 From: Pavan Kumar Date: Sun, 13 Oct 2024 03:33:18 +0100 Subject: [PATCH] add python operator imports to compat --- dev/tests_common/test_utils/compat.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/dev/tests_common/test_utils/compat.py b/dev/tests_common/test_utils/compat.py index 67a111350daff..6c0b96f1a56ca 100644 --- a/dev/tests_common/test_utils/compat.py +++ b/dev/tests_common/test_utils/compat.py @@ -60,6 +60,12 @@ from airflow.sensors.bash import BashSensor # type: ignore[no-redef,attr-defined] from airflow.sensors.date_time import DateTimeSensor # type: ignore[no-redef,attr-defined] +# Temporary imports here, once python operator moved to standard provider, this can be merged to above block +try: + from airflow.providers.standard.operators.python import PythonOperator +except ImportError: + from airflow.operators.python import PythonOperator # type: ignore[no-redef,attr-defined] + if TYPE_CHECKING: from airflow.models.asset import (