Skip to content

Commit b7d232c

Browse files
authored
[airflow] Add chain, chain_linear and cross_downstream for AIR302 (#16647)
## Summary Similar to #16014. PR on Airflow side: apache/airflow#47639 ## Test Plan A test fixture has been updated
1 parent 4d3a5af commit b7d232c

File tree

3 files changed

+505
-497
lines changed

3 files changed

+505
-497
lines changed

crates/ruff_linter/resources/test/fixtures/airflow/AIR302_names.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@
4444
from airflow.lineage.hook import DatasetLineageInfo
4545
from airflow.listeners.spec.dataset import on_dataset_changed, on_dataset_created
4646
from airflow.metrics.validators import AllowListValidator, BlockListValidator
47+
from airflow.models.baseoperator import chain, chain_linear, cross_downstream
4748
from airflow.operators import dummy_operator
4849
from airflow.operators.branch_operator import BaseBranchOperator
4950
from airflow.operators.dagrun_operator import TriggerDagRunLink, TriggerDagRunOperator

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

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -751,11 +751,18 @@ fn check_name(checker: &Checker, expr: &Expr, range: TextRange) {
751751
}
752752

753753
// airflow.utils.helpers
754-
["airflow", "utils", "helpers", "chain"] => {
755-
Replacement::Name("airflow.models.baseoperator.chain")
756-
}
754+
["airflow", "utils", "helpers", "chain"] => Replacement::Name("airflow.sdk.chain"),
757755
["airflow", "utils", "helpers", "cross_downstream"] => {
758-
Replacement::Name("airflow.models.baseoperator.cross_downstream")
756+
Replacement::Name("airflow.sdk.cross_downstream")
757+
}
758+
759+
// airflow.models.baseoperator
760+
["airflow", "models", "baseoperator", "chain"] => Replacement::Name("airflow.sdk.chain"),
761+
["airflow", "models", "baseoperator", "chain_linear"] => {
762+
Replacement::Name("airflow.sdk.chain_linear")
763+
}
764+
["airflow", "models", "baseoperator", "cross_downstream"] => {
765+
Replacement::Name("airflow.sdk.cross_downstream")
759766
}
760767

761768
// airflow.utils.state

0 commit comments

Comments
 (0)