networkx: type annotation needed for instantiating a graph

If I just do eg:
```python
    graph = nx.DiGraph()
```
`mypy` complains that
```
var-annotated Need type annotation for "graph"
```
Is there some issues with the stubs here that makes `mypy` not understand that `nx.DiGraph()` instantiates an instance of the class `DiGraph`?