await not allowed in comprehension in async def

[youknowone]

Feature

code

async def f():
    [await a for a in range(1)]

expected: No error

Actual:

  File "z.py", line 2
    [await a for a in range(1)]
     ^
SyntaxError: 'await' outside async function

Python Documentation or reference to CPython source code

[hydrogen602]

are you working on this right now? Otherwise I would like to take a stab at this and see if I get anywhere.

[youknowone]

Not yet, please go ahead. 👍
codegen/compile.rs is related.
propagating async function context to compile_comprehension will be the starting point