This is an automated email from the ASF dual-hosted git repository.
kaxilnaik 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 05df9925bef Provider Migration: Update Snowflake provider for Airflow
3.0 compatibility (#52629)
05df9925bef is described below
commit 05df9925bef7cea405feae048214595f90b23e9b
Author: Kalyan R <[email protected]>
AuthorDate: Tue Jul 1 23:16:49 2025 +0530
Provider Migration: Update Snowflake provider for Airflow 3.0
compatibility (#52629)
Part of #52378
---
.../providers/snowflake/transfers/copy_into_snowflake.py | 2 +-
.../src/airflow/providers/snowflake/version_compat.py | 10 ++++++++++
2 files changed, 11 insertions(+), 1 deletion(-)
diff --git
a/providers/snowflake/src/airflow/providers/snowflake/transfers/copy_into_snowflake.py
b/providers/snowflake/src/airflow/providers/snowflake/transfers/copy_into_snowflake.py
index e8b70344fbf..9032b9fbe5d 100644
---
a/providers/snowflake/src/airflow/providers/snowflake/transfers/copy_into_snowflake.py
+++
b/providers/snowflake/src/airflow/providers/snowflake/transfers/copy_into_snowflake.py
@@ -22,9 +22,9 @@ from __future__ import annotations
from collections.abc import Sequence
from typing import Any
-from airflow.models import BaseOperator
from airflow.providers.snowflake.hooks.snowflake import SnowflakeHook
from airflow.providers.snowflake.utils.common import enclose_param
+from airflow.providers.snowflake.version_compat import BaseOperator
def _validate_parameter(param_name: str, value: str | None) -> str | None:
diff --git
a/providers/snowflake/src/airflow/providers/snowflake/version_compat.py
b/providers/snowflake/src/airflow/providers/snowflake/version_compat.py
index 48d122b6696..df1b3e70378 100644
--- a/providers/snowflake/src/airflow/providers/snowflake/version_compat.py
+++ b/providers/snowflake/src/airflow/providers/snowflake/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 # type: ignore[no-redef]
+
+__all__ = [
+ "AIRFLOW_V_3_0_PLUS",
+ "BaseOperator",
+]
