This is an automated email from the ASF dual-hosted git repository.

potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new 75cb73f6ace Add poke_mode_only to version_compat to fix the incorrect 
deprecation warning (#56435)
75cb73f6ace is described below

commit 75cb73f6acea924a138dfca4d144c185193e6171
Author: Zach Liu <[email protected]>
AuthorDate: Mon Oct 13 17:24:22 2025 -0400

    Add poke_mode_only to version_compat to fix the incorrect deprecation 
warning (#56435)
    
    * fix the deprecation warning
    
    * for pre-3.1 versions
---
 providers/amazon/src/airflow/providers/amazon/aws/sensors/s3.py | 2 +-
 providers/amazon/src/airflow/providers/amazon/version_compat.py | 3 +++
 2 files changed, 4 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 c094b0d5ccf..42cdbd402ee 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.hooks.s3 import S3Hook
 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 f2e3fdfc6af..87cdd8c94d9 100644
--- a/providers/amazon/src/airflow/providers/amazon/version_compat.py
+++ b/providers/amazon/src/airflow/providers/amazon/version_compat.py
@@ -37,8 +37,10 @@ AIRFLOW_V_3_1_PLUS: bool = get_base_airflow_version_tuple() 
>= (3, 1, 0)
 
 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]
+    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
@@ -55,5 +57,6 @@ __all__ = [
     "BaseOperator",
     "BaseOperatorLink",
     "BaseSensorOperator",
+    "poke_mode_only",
     "XCom",
 ]

Reply via email to