From 8f7cf7afc9fe08ca248b4cfa435af765c05e4955 Mon Sep 17 00:00:00 2001 From: zachliu Date: Mon, 6 Oct 2025 10:46:24 -0400 Subject: [PATCH 1/2] fix the deprecation warning --- providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py | 2 +- providers/amazon/src/airflow/providers/amazon/version_compat.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py b/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py index c094b0d5ccfd9..42cdbd402ee19 100644 --- a/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py +++ b/providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py @@ -36,7 +36,7 @@ from airflow.providers.amazon.aws.sensors.base_aws import AwsBaseSensor from airflow.providers.amazon.aws.triggers.s3 import S3KeysUnchangedTrigger, S3KeyTrigger from airflow.providers.amazon.aws.utils.mixins import aws_template_fields -from airflow.sensors.base import poke_mode_only +from airflow.providers.amazon.version_compat import poke_mode_only class S3KeySensor(AwsBaseSensor[S3Hook]): diff --git a/providers/amazon/src/airflow/providers/amazon/version_compat.py b/providers/amazon/src/airflow/providers/amazon/version_compat.py index f2e3fdfc6afc5..47ce488d666a2 100644 --- a/providers/amazon/src/airflow/providers/amazon/version_compat.py +++ b/providers/amazon/src/airflow/providers/amazon/version_compat.py @@ -37,6 +37,7 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]: if AIRFLOW_V_3_1_PLUS: from airflow.sdk import BaseHook + from airflow.sdk.bases.sensor import poke_mode_only else: from airflow.hooks.base import BaseHook # type: ignore[attr-defined,no-redef] @@ -55,5 +56,6 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]: "BaseOperator", "BaseOperatorLink", "BaseSensorOperator", + "poke_mode_only", "XCom", ] From 7eeb07ebb6cdbfa09dc476285c05e004fe5e3258 Mon Sep 17 00:00:00 2001 From: zachliu Date: Mon, 6 Oct 2025 11:05:09 -0400 Subject: [PATCH 2/2] for pre-3.1 versions --- providers/amazon/src/airflow/providers/amazon/version_compat.py | 1 + 1 file changed, 1 insertion(+) diff --git a/providers/amazon/src/airflow/providers/amazon/version_compat.py b/providers/amazon/src/airflow/providers/amazon/version_compat.py index 47ce488d666a2..87cdd8c94d953 100644 --- a/providers/amazon/src/airflow/providers/amazon/version_compat.py +++ b/providers/amazon/src/airflow/providers/amazon/version_compat.py @@ -40,6 +40,7 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]: from airflow.sdk.bases.sensor import poke_mode_only else: from airflow.hooks.base import BaseHook # type: ignore[attr-defined,no-redef] + from airflow.sensors.base import poke_mode_only # type: ignore[no-redef] if AIRFLOW_V_3_0_PLUS: from airflow.sdk import BaseOperator, BaseOperatorLink, BaseSensorOperator