[networkx] Improve type coverage (#16028)

Author: yangdanny97

stubs/networkx/networkx/algorithms/approximation/distance_measures.pyi

@@ -5,4 +5,4 @@ from numpy.random import RandomState
 __all__ = ["diameter"]
 
 @_dispatchable
-def diameter(G: Graph[_Node], seed: int | RandomState | None = None): ...
+def diameter(G: Graph[_Node], seed: int | RandomState | None = None) -> int: ...

stubs/networkx/networkx/algorithms/approximation/maxcut.pyi

@@ -9,8 +9,8 @@ __all__ = ["randomized_partitioning", "one_exchange"]
 @_dispatchable
 def randomized_partitioning(
     G: Graph[_Node], seed: int | RandomState | None = None, p: float = 0.5, weight: str | None = None
-): ...
+) -> tuple[float, tuple[set[Incomplete], set[Incomplete]]]: ...
 @_dispatchable
 def one_exchange(
     G: Graph[_Node], initial_cut: set[Incomplete] | None = None, seed: int | RandomState | None = None, weight: str | None = None
-): ...
+) -> tuple[float, tuple[set[Incomplete], set[Incomplete]]]: ...

stubs/networkx/networkx/algorithms/approximation/ramsey.pyi

@@ -1,7 +1,9 @@
+from _typeshed import Incomplete
+
 from networkx.classes.graph import Graph, _Node
 from networkx.utils.backends import _dispatchable
 
 __all__ = ["ramsey_R2"]
 
 @_dispatchable
-def ramsey_R2(G: Graph[_Node]): ...
+def ramsey_R2(G: Graph[_Node]) -> tuple[set[Incomplete], set[Incomplete]]: ...

stubs/networkx/networkx/algorithms/approximation/steinertree.pyi

@@ -11,6 +11,8 @@ __all__ = ["metric_closure", "steiner_tree"]
 @deprecated(
     "`metric_closure` is deprecated and will be removed in NetworkX 3.8. Use `networkx.all_pairs_shortest_path_length` instead."
 )
-def metric_closure(G: Graph[_Node], weight="weight"): ...
+def metric_closure(G: Graph[_Node], weight="weight") -> Graph[Incomplete]: ...
 @_dispatchable
-def steiner_tree(G: Graph[_Node], terminal_nodes: Iterable[Incomplete], weight: str = "weight", method: str | None = None): ...
+def steiner_tree(
+    G: Graph[_Node], terminal_nodes: Iterable[Incomplete], weight: str = "weight", method: str | None = None
+) -> Graph[Incomplete]: ...

stubs/networkx/networkx/algorithms/approximation/traveling_salesman.pyi

@@ -30,40 +30,42 @@ def traveling_salesman_problem(
     cycle: bool = True,
     method: Callable[..., Incomplete] | None = None,
     **kwargs,
-): ...
+) -> list[Incomplete]: ...
 @_dispatchable
 def asadpour_atsp(
     G: DiGraph[_Node], weight: str | None = "weight", seed: int | RandomState | None = None, source: str | None = None
-): ...
+) -> list[Incomplete]: ...
 @_dispatchable
-def held_karp_ascent(G: Graph[_Node], weight="weight"): ...
+def held_karp_ascent(
+    G: Graph[_Node], weight: str = "weight"
+) -> tuple[float, dict[Incomplete, Incomplete] | Graph[Incomplete]]: ...
 @_dispatchable
 def spanning_tree_distribution(G: Graph[_Node], z: Mapping[Incomplete, Incomplete]) -> dict[Incomplete, Incomplete]: ...
 @_dispatchable
-def greedy_tsp(G: Graph[_Node], weight: str | None = "weight", source=None): ...
+def greedy_tsp(G: Graph[_Node], weight: str | None = "weight", source=None) -> list[Incomplete]: ...
 @_dispatchable
 def simulated_annealing_tsp(
     G: Graph[_Node],
-    init_cycle,
+    init_cycle: Literal["greedy"] | Iterable[Incomplete],
     weight: str | None = "weight",
     source=None,
     temp: int | None = 100,
-    move="1-1",
+    move: Callable[..., Incomplete] | Literal["1-1", "1-0"] = "1-1",
     max_iterations: int | None = 10,
     N_inner: int | None = 100,
-    alpha=0.01,
+    alpha: float = 0.01,
     seed: int | RandomState | None = None,
-): ...
+) -> list[Incomplete]: ...
 @_dispatchable
 def threshold_accepting_tsp(
     G: Graph[_Node],
     init_cycle: Literal["greedy"] | Iterable[Incomplete],
     weight: str | None = "weight",
     source=None,
     threshold: int | None = 1,
-    move="1-1",
+    move: Callable[..., Incomplete] | Literal["1-1", "1-0"] = "1-1",
     max_iterations: int | None = 10,
     N_inner: int | None = 100,
-    alpha=0.1,
+    alpha: float = 0.1,
     seed: int | RandomState | None = None,
-): ...
+) -> list[Incomplete]: ...

stubs/networkx/networkx/algorithms/assortativity/mixing.pyi

@@ -1,6 +1,7 @@
 from _typeshed import Incomplete
 from collections.abc import Iterable, Mapping
 
+import numpy as np
 from networkx.classes.graph import Graph, _Node
 from networkx.utils.backends import _dispatchable
 
@@ -17,7 +18,7 @@ def attribute_mixing_matrix(
     nodes: Iterable[Incomplete] | None = None,
     mapping: Mapping[Incomplete, Incomplete] | None = None,
     normalized: bool = True,
-): ...
+) -> np.ndarray[Incomplete, Incomplete]: ...
 @_dispatchable
 def degree_mixing_dict(
     G: Graph[_Node], x: str = "out", y: str = "in", weight: str | None = None, nodes=None, normalized: bool = False
@@ -31,6 +32,6 @@ def degree_mixing_matrix(
     nodes: Iterable[Incomplete] | None = None,
     normalized: bool = True,
     mapping: Mapping[Incomplete, Incomplete] | None = None,
-): ...
+) -> np.ndarray[Incomplete, Incomplete]: ...
 @_dispatchable
-def mixing_dict(xy, normalized: bool = False) -> dict[Incomplete, Incomplete]: ...
+def mixing_dict(xy: Iterable[tuple[Incomplete, Incomplete]], normalized: bool = False) -> dict[Incomplete, Incomplete]: ...

stubs/networkx/networkx/algorithms/bipartite/basic.pyi

@@ -1,5 +1,5 @@
 from _typeshed import Incomplete
-from collections.abc import Iterable
+from collections.abc import Collection, Iterable
 
 from networkx.classes.graph import Graph, _Node
 from networkx.utils.backends import _dispatchable
@@ -15,6 +15,6 @@ def is_bipartite_node_set(G: Graph[_Node], nodes: Iterable[Incomplete]) -> bool:
 @_dispatchable
 def sets(G: Graph[_Node], top_nodes: Iterable[Incomplete] | None = None) -> tuple[set[Incomplete], set[Incomplete]]: ...
 @_dispatchable
-def density(B: Graph[_Node], nodes) -> float: ...
+def density(B: Graph[_Node], nodes: Collection[Incomplete]) -> float: ...
 @_dispatchable
-def degrees(B: Graph[_Node], nodes, weight: str | None = None) -> tuple[Incomplete, Incomplete]: ...
+def degrees(B: Graph[_Node], nodes: Iterable[Incomplete], weight: str | None = None) -> tuple[Incomplete, Incomplete]: ...

stubs/networkx/networkx/algorithms/bipartite/centrality.pyi

@@ -1,13 +1,16 @@
 from _typeshed import Incomplete
+from collections.abc import Iterable
 
 from networkx.classes.graph import Graph, _Node
 from networkx.utils.backends import _dispatchable
 
 __all__ = ["degree_centrality", "betweenness_centrality", "closeness_centrality"]
 
 @_dispatchable
-def degree_centrality(G: Graph[_Node], nodes) -> dict[Incomplete, Incomplete]: ...
+def degree_centrality(G: Graph[_Node], nodes: Iterable[Incomplete]) -> dict[Incomplete, Incomplete]: ...
 @_dispatchable
-def betweenness_centrality(G: Graph[_Node], nodes) -> dict[Incomplete, Incomplete]: ...
+def betweenness_centrality(G: Graph[_Node], nodes: Iterable[Incomplete]) -> dict[Incomplete, Incomplete]: ...
 @_dispatchable
-def closeness_centrality(G: Graph[_Node], nodes, normalized: bool | None = True) -> dict[Incomplete, Incomplete]: ...
+def closeness_centrality(
+    G: Graph[_Node], nodes: Iterable[Incomplete], normalized: bool | None = True
+) -> dict[Incomplete, Incomplete]: ...

stubs/networkx/networkx/algorithms/bipartite/edgelist.pyi

@@ -1,4 +1,5 @@
-from collections.abc import Generator
+from _typeshed import Incomplete, StrPath, SupportsRead, SupportsWrite
+from collections.abc import Collection, Generator, Iterable
 
 from networkx.classes.graph import Graph, _Node
 from networkx.utils.backends import _dispatchable
@@ -7,27 +8,32 @@ __all__ = ["generate_edgelist", "write_edgelist", "parse_edgelist", "read_edgeli
 
 @_dispatchable
 def write_edgelist(
-    G: Graph[_Node], path, comments: str = "#", delimiter: str = " ", data: bool = True, encoding: str = "utf-8"
+    G: Graph[_Node],
+    path: StrPath | SupportsWrite[bytes],
+    comments: str = "#",
+    delimiter: str = " ",
+    data: bool = True,
+    encoding: str = "utf-8",
 ) -> None: ...
 @_dispatchable
 def generate_edgelist(G: Graph[_Node], delimiter: str = " ", data: bool = True) -> Generator[str]: ...
 @_dispatchable
 def parse_edgelist(
-    lines,
+    lines: Iterable[str],
     comments: str | None = "#",
     delimiter: str | None = None,
-    create_using: Graph[_Node] | None = None,
-    nodetype=None,
-    data=True,
-): ...
+    create_using: Graph[_Node] | type[Graph[_Node]] | None = None,
+    nodetype: type[Incomplete] | None = None,
+    data: bool | Collection[tuple[str, type[Incomplete]]] = True,
+) -> Graph[Incomplete]: ...
 @_dispatchable
 def read_edgelist(
-    path,
+    path: StrPath | SupportsRead[bytes],
     comments: str | None = "#",
     delimiter: str | None = None,
-    create_using=None,
+    create_using: Graph[Incomplete] | type[Graph[Incomplete]] | None = None,
     nodetype=None,
-    data=True,
+    data: bool | Collection[tuple[str, type[Incomplete]]] = True,
     edgetype=None,
     encoding: str | None = "utf-8",
-): ...
+) -> Graph[Incomplete]: ...

stubs/networkx/networkx/algorithms/bipartite/extendability.pyi

@@ -4,4 +4,4 @@ from networkx.utils.backends import _dispatchable
 __all__ = ["maximal_extendability"]
 
 @_dispatchable
-def maximal_extendability(G: Graph[_Node]): ...
+def maximal_extendability(G: Graph[_Node]) -> int: ...