mypy

Narrow context fallback limitation to walrus only

#21294

Merged