Resolve against the function, not the call

uranusjr · Nov 19, 2024 · fcd7936 · fcd7936
1 parent 8a72a7e
commit fcd7936
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/crates/ruff_linter/src/rules/airflow/rules/dag_schedule_argument.rs b/crates/ruff_linter/src/rules/airflow/rules/dag_schedule_argument.rs
@@ -64,15 +64,18 @@ impl Violation for AirflowDagNoScheduleArgument {
 /// AIR301
 pub(crate) fn dag_no_schedule_argument(checker: &mut Checker, expr: &Expr) {
     // Don't check non-call expressions.
-    let Expr::Call(ast::ExprCall { arguments, .. }) = expr else {
+    let Expr::Call(ast::ExprCall {
+        func, arguments, ..
+    }) = expr
+    else {
         return;
     };
 
     // We don't do anything unless this is a `DAG` (class) or `dag` (decorator
     // function) from Airflow.
     if !checker
         .semantic()
-        .resolve_qualified_name(expr)
+        .resolve_qualified_name(func)
         .is_some_and(|qualname| matches!(qualname.segments(), ["airflow", .., "DAG" | "dag"]))
     {
         return;