diff --git a/providers/redis/src/airflow/providers/redis/operators/redis_publish.py b/providers/redis/src/airflow/providers/redis/operators/redis_publish.py index fef0ff9bdd73d..dc933000cf565 100644 --- a/providers/redis/src/airflow/providers/redis/operators/redis_publish.py +++ b/providers/redis/src/airflow/providers/redis/operators/redis_publish.py @@ -20,8 +20,8 @@ from collections.abc import Sequence from typing import TYPE_CHECKING -from airflow.models import BaseOperator from airflow.providers.redis.hooks.redis import RedisHook +from airflow.providers.redis.version_compat import BaseOperator if TYPE_CHECKING: try: diff --git a/providers/redis/src/airflow/providers/redis/sensors/redis_key.py b/providers/redis/src/airflow/providers/redis/sensors/redis_key.py index 08f4e5ea0a65f..5f4eeb9fab454 100644 --- a/providers/redis/src/airflow/providers/redis/sensors/redis_key.py +++ b/providers/redis/src/airflow/providers/redis/sensors/redis_key.py @@ -21,12 +21,7 @@ from typing import TYPE_CHECKING from airflow.providers.redis.hooks.redis import RedisHook -from airflow.providers.redis.version_compat import AIRFLOW_V_3_0_PLUS - -if AIRFLOW_V_3_0_PLUS: - from airflow.sdk import BaseSensorOperator -else: - from airflow.sensors.base import BaseSensorOperator # type: ignore[no-redef] +from airflow.providers.redis.version_compat import BaseSensorOperator if TYPE_CHECKING: try: diff --git a/providers/redis/src/airflow/providers/redis/sensors/redis_pub_sub.py b/providers/redis/src/airflow/providers/redis/sensors/redis_pub_sub.py index 29e01ae6fa908..d2ba79d8246b1 100644 --- a/providers/redis/src/airflow/providers/redis/sensors/redis_pub_sub.py +++ b/providers/redis/src/airflow/providers/redis/sensors/redis_pub_sub.py @@ -22,12 +22,7 @@ from typing import TYPE_CHECKING from airflow.providers.redis.hooks.redis import RedisHook -from airflow.providers.redis.version_compat import AIRFLOW_V_3_0_PLUS - -if AIRFLOW_V_3_0_PLUS: - from airflow.sdk import BaseSensorOperator -else: - from airflow.sensors.base import BaseSensorOperator # type: ignore[no-redef] +from airflow.providers.redis.version_compat import BaseSensorOperator if TYPE_CHECKING: try: diff --git a/providers/redis/src/airflow/providers/redis/version_compat.py b/providers/redis/src/airflow/providers/redis/version_compat.py index 48d122b669696..765857caf8002 100644 --- a/providers/redis/src/airflow/providers/redis/version_compat.py +++ b/providers/redis/src/airflow/providers/redis/version_compat.py @@ -33,3 +33,11 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]: AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0) + +if AIRFLOW_V_3_0_PLUS: + from airflow.sdk import BaseOperator, BaseSensorOperator +else: + from airflow.models import BaseOperator # type: ignore[no-redef] + from airflow.sensors.base import BaseSensorOperator # type: ignore[no-redef] + +__all__ = ["BaseOperator", "BaseSensorOperator"]