Missing class in attention.pyi: MultiHeadAttention

1

Working with python3.7

In the terminal I can do: from torch.nn.modules.activation import MultiheadAttention

But in Pycharm, it says "cannot reference in ‘activation.pyi’. Indeed, I checked and MultiheadAttention exists in activation.py but not in activation.pyi.

Is that an error that should be reported?

2

Seem like a general issue. Feel free to open an issue or comment under the existing issue. If you open a new issue, please include a code snippet to reproduce the error.