Visit TypeVar and NewType name arguments

astral-sh · May 24, 2023 · 8289e3d · 8289e3d
1 parent 3644695
commit 8289e3d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/crates/ruff/src/checkers/ast/mod.rs b/crates/ruff/src/checkers/ast/mod.rs
@@ -4061,12 +4061,18 @@ where
                     }
                     Some(Callable::NewType) => {
                         self.visit_expr(func);
+                        if !args.is_empty() {
+                            self.visit_non_type_definition(&args[0]);
+                        }
                         for expr in args.iter().skip(1) {
                             self.visit_type_definition(expr);
                         }
                     }
                     Some(Callable::TypeVar) => {
                         self.visit_expr(func);
+                        if !args.is_empty() {
+                            self.visit_non_type_definition(&args[0]);
+                        }
                         for expr in args.iter().skip(1) {
                             self.visit_type_definition(expr);
                         }