From 84f8e63a2b023539f01a80e936674269d4bf3fc7 Mon Sep 17 00:00:00 2001 From: Axel Dahlberg Date: Wed, 9 Jul 2025 10:21:33 +0200 Subject: [PATCH] networkx: fix subgraph and degree --- stubs/networkx/networkx/classes/graph.pyi | 2 +- stubs/networkx/networkx/classes/reportviews.pyi | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/stubs/networkx/networkx/classes/graph.pyi b/stubs/networkx/networkx/classes/graph.pyi index ec771d64ece9..c18dbbe81cff 100644 --- a/stubs/networkx/networkx/classes/graph.pyi +++ b/stubs/networkx/networkx/classes/graph.pyi @@ -90,7 +90,7 @@ class Graph(Collection[_Node]): def copy(self, as_view: bool = False) -> Self: ... def to_directed(self, as_view: bool = False) -> DiGraph[_Node]: ... def to_undirected(self, as_view: bool = False) -> Graph[_Node]: ... - def subgraph(self, nodes: Iterable[_Node]) -> Graph[_Node]: ... + def subgraph(self, nodes: _NBunch[_Node]) -> Graph[_Node]: ... def edge_subgraph(self, edges: Iterable[_Edge[_Node]]) -> Graph[_Node]: ... @overload def size(self, weight: None = None) -> int: ... diff --git a/stubs/networkx/networkx/classes/reportviews.pyi b/stubs/networkx/networkx/classes/reportviews.pyi index ff9078493a19..87e04be2b0e2 100644 --- a/stubs/networkx/networkx/classes/reportviews.pyi +++ b/stubs/networkx/networkx/classes/reportviews.pyi @@ -55,7 +55,12 @@ class NodeDataView(AbstractSet[_Node]): class DiDegreeView(Generic[_Node]): def __init__(self, G: Graph[_Node], nbunch: _NBunch[_Node] = None, weight: None | bool | str = None) -> None: ... - def __call__(self, nbunch: _NBunch[_Node] = None, weight: None | bool | str = None) -> int | DiDegreeView[_Node]: ... + @overload + def __call__(self, nbunch: None = None, weight: None | bool | str = None) -> Self: ... + @overload + def __call__(self, nbunch: _Node, weight: None | bool | str = None) -> int: ... + @overload + def __call__(self, nbunch: Iterable[_Node], weight: None | bool | str = None) -> Self: ... def __getitem__(self, n: _Node) -> float: ... def __iter__(self) -> Iterator[tuple[_Node, float]]: ... def __len__(self) -> int: ...