[ty] Recognise functions containing yield from expressions as being generator functions (#17930)

2025-05-07 23:29:44 +01:00
parent 2cf5cba7ff
commit 51cef5a72b
3 changed files with 30 additions and 24 deletions
--- a/crates/ty_python_semantic/src/semantic_index/builder.rs
+++ b/crates/ty_python_semantic/src/semantic_index/builder.rs
@@ -2310,7 +2310,7 @@ where

                walk_expr(self, expr);
            }
-            ast::Expr::Yield(_) => {
+            ast::Expr::Yield(_) | ast::Expr::YieldFrom(_) => {
                let scope = self.current_scope();
                if self.scopes[scope].kind() == ScopeKind::Function {
                    self.generator_functions.insert(scope);