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 (