Skip to content

Commit 60f780e

Browse files
authored
Replace models.BaseOperator to Task SDK one for Mongo (apache#52566)
1 parent 3316407 commit 60f780e

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

providers/mongo/src/airflow/providers/mongo/sensors/mongo.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,7 @@
2121
from typing import TYPE_CHECKING
2222

2323
from airflow.providers.mongo.hooks.mongo import MongoHook
24-
from airflow.providers.mongo.version_compat import AIRFLOW_V_3_0_PLUS
25-
26-
if AIRFLOW_V_3_0_PLUS:
27-
from airflow.sdk import BaseSensorOperator
28-
else:
29-
from airflow.sensors.base import BaseSensorOperator # type: ignore[no-redef]
24+
from airflow.providers.mongo.version_compat import BaseSensorOperator
3025

3126
if TYPE_CHECKING:
3227
try:

providers/mongo/src/airflow/providers/mongo/version_compat.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,3 +33,10 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]:
3333

3434

3535
AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0)
36+
37+
if AIRFLOW_V_3_0_PLUS:
38+
from airflow.sdk import BaseSensorOperator
39+
else:
40+
from airflow.sensors.base import BaseSensorOperator # type: ignore[no-redef]
41+
42+
__all__ = ["AIRFLOW_V_3_0_PLUS", "BaseSensorOperator"]

0 commit comments

Comments
 (0)