Allow walrus in slices

See python/cpython#23317 Raised in #930.
Instagram · May 27, 2023 · 110109a · 110109a
1 parent 59aeceb
commit 110109a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/native/libcst/src/parser/grammar.rs b/native/libcst/src/parser/grammar.rs
@@ -1004,7 +1004,7 @@ parser! {
                     make_slice(l, col, u, rest)
             }
             / e:starred_expression() { make_index_from_arg(e) }
-            / v:expression() { make_index(v) }
+            / v:named_expression() { make_index(v) }
 
         rule atom() -> Expression<'input, 'a>
             = n:name() { Expression::Name(Box::new(n)) }

diff --git a/native/libcst/tests/fixtures/wonky_walrus.py b/native/libcst/tests/fixtures/wonky_walrus.py
@@ -10,4 +10,6 @@
 if f := x(): pass
 
 f(y:=1)
-f(x,  y   :=    1     )
+f(x,  y   :=    1     )
+
+_[_:=10]