ipython / ipython/ipykernel

ContextVar not preserved when cell contains `await`

Open
#749 3 comments 2 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
734
Forks
412
Avg merge
1d 5h
Merged PRs (30d)
8

Description

See https://gist.github.com/mvolfik/2582939ec00120a04503507214c24105

```python3
import contextvars, asyncio
VAR = contextvars.ContextVar("VAR", default="default")

async def produce_result(x):
return x

async def read_var():
print("Var:", VAR.get())

# The following is broken

# In[3]:
result = await produce_result("bug fixed yayy")
VAR.set(result)

# In[4]:
await read_var()
# Var: default

# The following works

# In[5]:
result = await produce_result("this worked before")

# In[6]:
VAR.set(result)

# In[7]:
await read_var()
# Var: this worked before
```

In the working case, `await` result is stored in a variable, and `VAR` is set from cell which is fully synchronous (idk ipython internals, but that's what I guess makes the difference). What breaks is setting var from the same cell where we call and await the async function.

I'm using `ipykernel==6.2.0`. Neither of these cases worked (both outputted 'default') with `ipykernel==5.5.0`.

Related to ipython issue ipython/ipython#11565, which was fixed by #632

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.