Merge branch 'enhance-visualization-typing' of github.com:zen-xu/rustworkx into enhance-visualization-typing

Author: zen-xu

rustworkx/rustworkx.pyi

@@ -1240,7 +1240,7 @@ class PyGraph(Generic[_S, _T]):
         /,
         weight_combo_fn: Callable[[_T, _T], _T] | None = ...,
     ) -> int: ...
-    def copy(self) -> PyGraph[_S, _T]: ...
+    def copy(self) -> Self: ...
     def degree(self, node: int, /) -> int: ...
     def edge_index_map(self) -> EdgeIndexMap[_T]: ...
     def edge_indices(self) -> EdgeIndices: ...
@@ -1314,14 +1314,24 @@ class PyGraph(Generic[_S, _T]):
         node_filter: Callable[[_S], bool] | None = ...,
         edge_weight_map: Callable[[_T], _T] | None = ...,
     ) -> NodeMap: ...
+    @overload
     def to_dot(
         self,
         /,
         node_attr: Callable[[_S], dict[str, str]] | None = ...,
         edge_attr: Callable[[_T], dict[str, str]] | None = ...,
         graph_attr: dict[str, str] | None = ...,
-        filename: str | None = None,
-    ) -> str | None: ...
+        filename: None = ...,
+    ) -> str: ...
+    @overload
+    def to_dot(
+        self,
+        /,
+        node_attr: Callable[[_S], dict[str, str]] | None = ...,
+        edge_attr: Callable[[_T], dict[str, str]] | None = ...,
+        graph_attr: dict[str, str] | None = ...,
+        filename: str = ...,
+    ) -> None: ...
     def to_directed(self) -> PyDiGraph[_S, _T]: ...
     def update_edge(
         self,
@@ -1398,7 +1408,7 @@ class PyDiGraph(Generic[_S, _T]):
         check_cycle: bool | None = ...,
         weight_combo_fn: Callable[[_T, _T], _T] | None = ...,
     ) -> int: ...
-    def copy(self) -> PyDiGraph[_S, _T]: ...
+    def copy(self) -> Self: ...
     def edge_index_map(self) -> EdgeIndexMap[_T]: ...
     def edge_indices(self) -> EdgeIndices: ...
     def edge_indices_from_endpoints(self, node_a: int, node_b: int) -> EdgeIndices: ...
@@ -1514,14 +1524,24 @@ class PyDiGraph(Generic[_S, _T]):
     ) -> NodeMap: ...
     def successor_indices(self, node: int, /) -> NodeIndices: ...
     def successors(self, node: int, /) -> list[_S]: ...
+    @overload
     def to_dot(
         self,
         /,
         node_attr: Callable[[_S], dict[str, str]] | None = ...,
         edge_attr: Callable[[_T], dict[str, str]] | None = ...,
         graph_attr: dict[str, str] | None = ...,
-        filename: str | None = None,
-    ) -> str | None: ...
+        filename: None = ...,
+    ) -> str: ...
+    @overload
+    def to_dot(
+        self,
+        /,
+        node_attr: Callable[[_S], dict[str, str]] | None = ...,
+        edge_attr: Callable[[_T], dict[str, str]] | None = ...,
+        graph_attr: dict[str, str] | None = ...,
+        filename: str = ...,
+    ) -> None: ...
     def to_undirected(
         self,
         /,