From 79962d5d14bb0904abaf6826ed68e476578a7a12 Mon Sep 17 00:00:00 2001 From: Dov Benyomin Sohacheski Date: Mon, 30 Jun 2025 07:56:54 +0300 Subject: [PATCH] Replace models.BaseOperator to Task SDK for apache/hdfs --- .../src/airflow/providers/apache/hdfs/hooks/webhdfs.py | 2 -- .../src/airflow/providers/apache/hdfs/sensors/web_hdfs.py | 7 +------ .../src/airflow/providers/apache/hdfs/version_compat.py | 7 +++++++ 3 files changed, 8 insertions(+), 8 deletions(-) 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"]