Skip to content

Commit c901b9b

Browse files
authored
Correct type annotations on NetworkX DiGraphs (#14595)
1 parent 4f82a8d commit c901b9b

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

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: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,10 @@ 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) -> Self: ...
4545
@overload
46-
def __call__(self, data: Literal[True] | str, default=None) -> Iterator[tuple[_Node, dict[str, Any]]]: ...
47-
def data(self, data: bool | str = True, default=None) -> NodeDataView[_Node]: ...
46+
def __call__(self, data: Literal[True] | str, default=None) -> Self: ...
47+
def data(self, data: bool | str = True, default=None) -> Self: ...
4848

4949
class NodeDataView(AbstractSet[_Node]):
5050
def __init__(self, nodedict: Mapping[str, Incomplete], data: bool | str = False, default=None) -> None: ...
@@ -55,7 +55,10 @@ class NodeDataView(AbstractSet[_Node]):
5555

5656
class DiDegreeView(Generic[_Node]):
5757
def __init__(self, G: Graph[_Node], nbunch: _NBunch[_Node] = None, weight: None | bool | str = None) -> None: ...
58-
def __call__(self, nbunch: _NBunch[_Node] = None, weight: None | bool | str = None) -> int | DiDegreeView[_Node]: ...
58+
@overload
59+
def __call__(self, nbunch: None = None, weight: None | bool | str = None) -> int: ... # type: ignore[overload-overlap]
60+
@overload
61+
def __call__(self, nbunch: None | Iterable[_Node], weight: None | bool | str = None) -> Self: ...
5962
def __getitem__(self, n: _Node) -> float: ...
6063
def __iter__(self) -> Iterator[tuple[_Node, float]]: ...
6164
def __len__(self) -> int: ...

0 commit comments

Comments
 (0)