From 2480d7e7c74493a024eaf254c5d2c6f452c80ee2 Mon Sep 17 00:00:00 2001
From: GiGaGon <107241144+MeGaGiGaGon@users.noreply.github.com>
Date: Sun, 31 Aug 2025 07:42:48 -0700
Subject: [PATCH] networkx: Fix incompletes in connectivity/connectivity
 (#14661)

---
 .../algorithms/connectivity/connectivity.pyi  | 30 +++++++++++--------
 1 file changed, 18 insertions(+), 12 deletions(-)

diff --git a/stubs/networkx/networkx/algorithms/connectivity/connectivity.pyi b/stubs/networkx/networkx/algorithms/connectivity/connectivity.pyi
index ca37ffb15..e38e1c82f 100644
--- a/stubs/networkx/networkx/algorithms/connectivity/connectivity.pyi
+++ b/stubs/networkx/networkx/algorithms/connectivity/connectivity.pyi
@@ -1,4 +1,3 @@
-from _typeshed import Incomplete
 from collections.abc import Callable, Iterable
 
 from networkx.algorithms.flow import edmonds_karp
@@ -21,36 +20,43 @@ def local_node_connectivity(
     G: Graph[_Node],
     s: _Node,
     t: _Node,
-    flow_func: Callable[..., Incomplete] | None = None,
+    flow_func: Callable[[DiGraph[_Node], _Node, _Node], DiGraph[_Node]] | None = None,
     auxiliary: DiGraph[_Node] | None = None,
     residual: DiGraph[_Node] | None = None,
     cutoff: float | None = None,
-): ...
+) -> float: ...
 @_dispatchable
 def node_connectivity(
-    G: Graph[_Node], s: _Node | None = None, t: _Node | None = None, flow_func: Callable[..., Incomplete] | None = None
-): ...
+    G: Graph[_Node],
+    s: _Node | None = None,
+    t: _Node | None = None,
+    flow_func: Callable[[DiGraph[_Node], _Node, _Node], DiGraph[_Node]] | None = None,
+) -> float: ...
 @_dispatchable
-def average_node_connectivity(G: Graph[_Node], flow_func: Callable[..., Incomplete] | None = None) -> float: ...
+def average_node_connectivity(
+    G: Graph[_Node], flow_func: Callable[[DiGraph[_Node], _Node, _Node], DiGraph[_Node]] | None = None
+) -> float: ...
 @_dispatchable
 def all_pairs_node_connectivity(
-    G: Graph[_Node], nbunch: Iterable[Incomplete] | None = None, flow_func: Callable[..., Incomplete] | None = None
-) -> dict[Incomplete, dict[Incomplete, Incomplete]]: ...
+    G: Graph[_Node],
+    nbunch: Iterable[tuple[_Node, _Node]] | None = None,
+    flow_func: Callable[[DiGraph[_Node], _Node, _Node], DiGraph[_Node]] | None = None,
+) -> dict[_Node, dict[_Node, float]]: ...
 @_dispatchable
 def local_edge_connectivity(
     G: Graph[_Node],
     s: _Node,
     t: _Node,
-    flow_func: Callable[..., Incomplete] | None = None,
+    flow_func: Callable[[DiGraph[_Node], _Node, _Node], DiGraph[_Node]] | None = None,
     auxiliary: DiGraph[_Node] | None = None,
     residual: DiGraph[_Node] | None = None,
     cutoff: float | None = None,
-): ...
+) -> float: ...
 @_dispatchable
 def edge_connectivity(
     G: Graph[_Node],
     s: _Node | None = None,
     t: _Node | None = None,
-    flow_func: Callable[..., Incomplete] | None = None,
+    flow_func: Callable[[DiGraph[_Node], _Node, _Node], DiGraph[_Node]] | None = None,
     cutoff: float | None = None,
-): ...
+) -> float: ...