Typing Hints for Inheritance

[patschmidt2]

Hello,

I am trying to use the networkx-types package to annotate the specific nodes I am using on my graph. Specifically, I want to inherit from the nx.DiGraph class with type hints, which should look something like this:

class MyGraph(nx.DiGraph[NodeType]):
    ...

However, this fails with TypeError: type 'DiGraph' is not subscriptable. What is the correct way to implement this here?
Thanks in advance!

[Daraan]

This can happen is there is a stub package for type-hints, but the actual runtime implementation is non-generic. You can work around this by defining your child class like this:

from typing import TYPE_CHECKING

class MyGraph(nx.DiGraph[NodeType] if TYPE_CHECKING else nx.DiGraph):
    ...

[patschmidt2]

Yes, that was exactly the case here! Thank you!

[Daraan]

Glad I could help. Feel free to also mark it as an answer :)