Skip to content

Commit 11d4346

Browse files
committed
enhance mpl_draw type hint
1 parent 6501d32 commit 11d4346

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

rustworkx/visualization/matplotlib.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ if typing.TYPE_CHECKING:
2020
_S = typing.TypeVar("_S")
2121
_T = typing.TypeVar("_T")
2222

23-
class _DrawKwargs(typing.TypedDict, total=False):
23+
class _DrawKwargs(typing.TypedDict, typing.Generic[_S, _T], total=False):
2424
arrowstyle: str
2525
arrow_size: int
2626
node_list: list[int]
@@ -67,5 +67,5 @@ def mpl_draw(
6767
ax: Axes | None = ...,
6868
arrows: bool = ...,
6969
with_labels: bool = ...,
70-
**kwds: typing_extensions.Unpack[_DrawKwargs],
70+
**kwds: typing_extensions.Unpack[_DrawKwargs[_S, _T]],
7171
) -> Figure | None: ...

0 commit comments

Comments
 (0)