From eb1aed121a228781d40840a191534a8dd575fd82 Mon Sep 17 00:00:00 2001 From: kyungjunleeme Date: Sat, 28 Jun 2025 14:28:50 +0900 Subject: [PATCH 1/2] ADD: impport for BaseOperator in version_compat.py --- .../src/airflow/providers/trino/version_compat.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/providers/trino/src/airflow/providers/trino/version_compat.py b/providers/trino/src/airflow/providers/trino/version_compat.py index 48d122b669696..e7a259afb357c 100644 --- a/providers/trino/src/airflow/providers/trino/version_compat.py +++ b/providers/trino/src/airflow/providers/trino/version_compat.py @@ -33,3 +33,13 @@ 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 +else: + from airflow.models import BaseOperator + +__all__ = [ + "AIRFLOW_V_3_0_PLUS", + "BaseOperator", +] From 89d6ffe4e9bdab5c6c508c6fbb3ae5c5df5675b3 Mon Sep 17 00:00:00 2001 From: kyungjunleeme Date: Sat, 28 Jun 2025 14:32:37 +0900 Subject: [PATCH 2/2] CHG: import change --- .../trino/src/airflow/providers/trino/transfers/gcs_to_trino.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/trino/src/airflow/providers/trino/transfers/gcs_to_trino.py b/providers/trino/src/airflow/providers/trino/transfers/gcs_to_trino.py index 65e8b2d6d9e59..ffa09b4268b4d 100644 --- a/providers/trino/src/airflow/providers/trino/transfers/gcs_to_trino.py +++ b/providers/trino/src/airflow/providers/trino/transfers/gcs_to_trino.py @@ -25,9 +25,9 @@ from tempfile import NamedTemporaryFile from typing import TYPE_CHECKING -from airflow.models import BaseOperator from airflow.providers.google.cloud.hooks.gcs import GCSHook from airflow.providers.trino.hooks.trino import TrinoHook +from airflow.providers.trino.version_compat import BaseOperator if TYPE_CHECKING: try: