Skip to content

Commit 0a88188

Browse files
committed
move BaseOperator to airflow.sdk
1 parent f1db649 commit 0a88188

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

crates/ruff_linter/src/rules/airflow/rules/removal_in_3_1.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ fn check_name(checker: &Checker, expr: &Expr, range: TextRange) {
168168
},
169169
// airflow.models.baseoperator
170170
["airflow", "models", "baseoperator", "BaseOperator"] => Replacement::SourceModuleMoved {
171-
module: "airflow.sdk.bases.operator",
171+
module: "airflow.sdk",
172172
name: "BaseOperator".to_string(),
173173
},
174174
// airflow.macros

crates/ruff_linter/src/rules/airflow/snapshots/ruff_linter__rules__airflow__tests__AIR321_AIR321_names.py.snap

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -379,12 +379,12 @@ AIR321 [*] `airflow.models.baseoperator.BaseOperator` is removed in Airflow 3.1
379379
68 |
380380
69 | # airflow.macros
381381
|
382-
help: Use `BaseOperator` from `airflow.sdk.bases.operator` instead.
382+
help: Use `BaseOperator` from `airflow.sdk` instead.
383383
62 | TaskStateChangeCallback()
384384
63 |
385385
64 | # airflow.models.baseoperator
386386
- from airflow.models.baseoperator import BaseOperator
387-
65 + from airflow.sdk.bases.operator import BaseOperator
387+
65 + from airflow.sdk import BaseOperator
388388
66 |
389389
67 | BaseOperator()
390390
68 |

0 commit comments

Comments
 (0)