Miscellaneous type annotation fixes

Qiskit · Jan 8, 2025 · a9136cd · a9136cd
1 parent 752555d
commit a9136cd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/rustworkx/__init__.pyi b/rustworkx/__init__.pyi
@@ -453,7 +453,7 @@ def spring_layout(
 def networkx_converter(graph: Any, keep_attributes: bool = ...) -> PyGraph | PyDiGraph: ...
 def bipartite_layout(
     graph: PyGraph[_S, _T] | PyDiGraph[_S, _T],
-    first_nodes,
+    first_nodes: set[int],
     horizontal: bool = ...,
     scale: int = ...,
     center: tuple[float, float] | None = ...,
@@ -588,7 +588,7 @@ def dijkstra_search(
 ) -> None: ...
 def bellman_ford_shortest_paths(
     graph: PyGraph[_S, _T] | PyDiGraph[_S, _T],
-    source,
+    source: int,
     target: int | None = ...,
     weight_fn: Callable[[_T], float] | None = ...,
     default_weight: float = ...,

diff --git a/rustworkx/rustworkx.pyi b/rustworkx/rustworkx.pyi
@@ -1307,7 +1307,7 @@ class PyGraph(Generic[_S, _T]):
     def __getstate__(self) -> Any: ...
     def __len__(self) -> int: ...
     def __setitem__(self, idx: int, value: _S, /) -> None: ...
-    def __setstate__(self, state, /) -> None: ...
+    def __setstate__(self, state: Any, /) -> None: ...
 
 # Digraph
 
@@ -1508,4 +1508,4 @@ class PyDiGraph(Generic[_S, _T]):
     def __getstate__(self) -> Any: ...
     def __len__(self) -> int: ...
     def __setitem__(self, idx: int, value: _S, /) -> None: ...
-    def __setstate__(self, state, /) -> None: ...
+    def __setstate__(self, state: Any, /) -> None: ...