[airflow] Add chain, chain_linear and cross_downstream for AIR302 (#16647)
## Summary Similar to https://github.com/astral-sh/ruff/pull/16014. PR on Airflow side: https://github.com/apache/airflow/pull/47639 ## Test Plan A test fixture has been updated
This commit is contained in:
@@ -44,6 +44,7 @@ from airflow.hooks.base_hook import BaseHook
|
||||
from airflow.lineage.hook import DatasetLineageInfo
|
||||
from airflow.listeners.spec.dataset import on_dataset_changed, on_dataset_created
|
||||
from airflow.metrics.validators import AllowListValidator, BlockListValidator
|
||||
from airflow.models.baseoperator import chain, chain_linear, cross_downstream
|
||||
from airflow.operators import dummy_operator
|
||||
from airflow.operators.branch_operator import BaseBranchOperator
|
||||
from airflow.operators.dagrun_operator import TriggerDagRunLink, TriggerDagRunOperator
|
||||
|
||||
@@ -751,11 +751,18 @@ fn check_name(checker: &Checker, expr: &Expr, range: TextRange) {
|
||||
}
|
||||
|
||||
// airflow.utils.helpers
|
||||
["airflow", "utils", "helpers", "chain"] => {
|
||||
Replacement::Name("airflow.models.baseoperator.chain")
|
||||
}
|
||||
["airflow", "utils", "helpers", "chain"] => Replacement::Name("airflow.sdk.chain"),
|
||||
["airflow", "utils", "helpers", "cross_downstream"] => {
|
||||
Replacement::Name("airflow.models.baseoperator.cross_downstream")
|
||||
Replacement::Name("airflow.sdk.cross_downstream")
|
||||
}
|
||||
|
||||
// airflow.models.baseoperator
|
||||
["airflow", "models", "baseoperator", "chain"] => Replacement::Name("airflow.sdk.chain"),
|
||||
["airflow", "models", "baseoperator", "chain_linear"] => {
|
||||
Replacement::Name("airflow.sdk.chain_linear")
|
||||
}
|
||||
["airflow", "models", "baseoperator", "cross_downstream"] => {
|
||||
Replacement::Name("airflow.sdk.cross_downstream")
|
||||
}
|
||||
|
||||
// airflow.utils.state
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user