Skip to content

Commit cd2e2c4

Browse files
authored
Correct type annotations on NetworkX NodeViews
1 parent 8a56044 commit cd2e2c4

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

stubs/networkx/networkx/classes/digraph.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,9 @@ class DiGraph(Graph[_Node]):
3434
@cached_property
3535
def in_edges(self) -> InEdgeView[_Node]: ...
3636
@cached_property
37-
def in_degree(self) -> int | InDegreeView[_Node] | InMultiDegreeView[_Node]: ...
37+
def in_degree(self) -> InDegreeView[_Node] | InMultiDegreeView[_Node]: ...
3838
@cached_property
39-
def out_degree(self) -> int | OutDegreeView[_Node] | OutMultiDegreeView[_Node]: ...
39+
def out_degree(self) -> OutDegreeView[_Node] | OutMultiDegreeView[_Node]: ...
4040
def to_undirected(self, reciprocal: bool = False, as_view: bool = False) -> Graph[_Node]: ... # type: ignore[override]
4141
# reciprocal : If True, only edges that appear in both directions ... will be kept in the undirected graph.
4242
def reverse(self, copy: bool = True) -> Self: ...

stubs/networkx/networkx/classes/reportviews.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,9 @@ class NodeView(Mapping[_Node, dict[str, Any]], AbstractSet[_Node]):
4141
def __getitem__(self, n: _Node) -> dict[str, Any]: ...
4242
def __contains__(self, n: object) -> bool: ...
4343
@overload
44-
def __call__(self, data: Literal[False] = False, default=None) -> Iterator[_Node]: ...
44+
def __call__(self, data: Literal[False] = False, default=None) -> NodeView[_Node]: ...
4545
@overload
46-
def __call__(self, data: Literal[True] | str, default=None) -> Iterator[tuple[_Node, dict[str, Any]]]: ...
46+
def __call__(self, data: Literal[True] | str, default=None) -> NodeDataView[_Node]: ...
4747
def data(self, data: bool | str = True, default=None) -> NodeDataView[_Node]: ...
4848

4949
class NodeDataView(AbstractSet[_Node]):

0 commit comments

Comments
 (0)