From 472d664c95375bdc13be65fb705f7ac106f47ba5 Mon Sep 17 00:00:00 2001
From: Anders Kaseorg <andersk@mit.edu>
Date: Tue, 27 Dec 2022 12:13:50 -0800
Subject: [PATCH] Allow named expression in set comprehension: `{a := b for c
 in d}`

Signed-off-by: Anders Kaseorg <andersk@mit.edu>
---
 parser/python.lalrpop | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/parser/python.lalrpop b/parser/python.lalrpop
index 4ec31c0..a49379c 100644
--- a/parser/python.lalrpop
+++ b/parser/python.lalrpop
@@ -1185,7 +1185,7 @@ Atom<Goal>: ast::Expr = {
         custom: (),
         node: ast::ExprKind::Set { elts }
     },
-    <location:@L> "{" <elt:Test<"all">> <generators:CompFor> "}" <end_location:@R> => {
+    <location:@L> "{" <elt:NamedExpressionTest> <generators:CompFor> "}" <end_location:@R> => {
         ast::Expr {
             location,
             end_location: Some(end_location),