From 84f8e63a2b023539f01a80e936674269d4bf3fc7 Mon Sep 17 00:00:00 2001
From: Axel Dahlberg <git@valleymnt.com>
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: ...