diff --git a/providers/apache/hdfs/src/airflow/providers/apache/hdfs/hooks/webhdfs.py b/providers/apache/hdfs/src/airflow/providers/apache/hdfs/hooks/webhdfs.py index f84588df54fe5..e3b735228fd2f 100644 --- a/providers/apache/hdfs/src/airflow/providers/apache/hdfs/hooks/webhdfs.py +++ b/providers/apache/hdfs/src/airflow/providers/apache/hdfs/hooks/webhdfs.py @@ -15,8 +15,6 @@ # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. -"""Hook for Web HDFS.""" - from __future__ import annotations import logging diff --git a/providers/apache/hdfs/src/airflow/providers/apache/hdfs/sensors/web_hdfs.py b/providers/apache/hdfs/src/airflow/providers/apache/hdfs/sensors/web_hdfs.py index cbd60378a5cfe..342cb5b526ba6 100644 --- a/providers/apache/hdfs/src/airflow/providers/apache/hdfs/sensors/web_hdfs.py +++ b/providers/apache/hdfs/src/airflow/providers/apache/hdfs/sensors/web_hdfs.py @@ -20,12 +20,7 @@ from collections.abc import Sequence from typing import TYPE_CHECKING, Any -from airflow.providers.apache.hdfs.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.apache.hdfs.version_compat import BaseSensorOperator if TYPE_CHECKING: from hdfs import InsecureClient diff --git a/providers/apache/hdfs/src/airflow/providers/apache/hdfs/version_compat.py b/providers/apache/hdfs/src/airflow/providers/apache/hdfs/version_compat.py index 48d122b669696..319d86aca697c 100644 --- a/providers/apache/hdfs/src/airflow/providers/apache/hdfs/version_compat.py +++ b/providers/apache/hdfs/src/airflow/providers/apache/hdfs/version_compat.py @@ -33,3 +33,10 @@ 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 BaseSensorOperator +else: + from airflow.sensors.base import BaseSensorOperator # type: ignore[no-redef] + +__all__ = ["AIRFLOW_V_3_0_PLUS", "BaseSensorOperator"]