diff --git a/tanjun/abc.py b/tanjun/abc.py index c8ce216a9..cee0f85d1 100644 --- a/tanjun/abc.py +++ b/tanjun/abc.py @@ -51,8 +51,8 @@ "HookSig", "Hooks", "ListenerCallbackSig", - "MenuCommand", "MenuCallbackSig", + "MenuCommand", "MenuContext", "MenuHooks", "MessageCommand", diff --git a/tanjun/commands/menu.py b/tanjun/commands/menu.py index 1e9ca6c51..d8568b635 100644 --- a/tanjun/commands/menu.py +++ b/tanjun/commands/menu.py @@ -74,9 +74,7 @@ def __call__(self, _: _AnyCommandT[_AnyCallbackSigT], /) -> MenuCommand[_AnyCall def __call__(self, _: _MenuCommandCallbackSigT, /) -> MenuCommand[_MenuCommandCallbackSigT, _MenuTypeT]: ... - def __call__( - self, _: _CallbackishT[_AnyCallbackSigT], / - ) -> MenuCommand[_AnyCallbackSigT, _MenuTypeT]: + def __call__(self, _: _CallbackishT[_AnyCallbackSigT], /) -> MenuCommand[_AnyCallbackSigT, _MenuTypeT]: raise NotImplementedError @@ -97,9 +95,7 @@ def decorator(callback: _AnyCommandT[_AnyCallbackSigT], /) -> MenuCommand[_AnyCa def decorator(callback: _MenuCommandCallbackSigT, /) -> MenuCommand[_MenuCommandCallbackSigT, _MenuTypeT]: ... - def decorator( - callback: _CallbackishT[_AnyCallbackSigT], / - ) -> MenuCommand[_AnyCallbackSigT, _MenuTypeT]: + def decorator(callback: _CallbackishT[_AnyCallbackSigT], /) -> MenuCommand[_AnyCallbackSigT, _MenuTypeT]: if isinstance(callback, (tanjun.MenuCommand, tanjun.MessageCommand, tanjun.SlashCommand)): wrapped_command = callback callback = callback.callback @@ -323,7 +319,7 @@ class MenuCommand(base.PartialCommand[tanjun.MenuContext], tanjun.MenuCommand[_M @typing.overload def __init__( - self: MenuCommand[_AnyCallbackSigT,_MenuTypeT], + self: MenuCommand[_AnyCallbackSigT, _MenuTypeT], callback: _AnyCommandT[ _AnyCallbackSigT, ], @@ -358,7 +354,7 @@ def __init__( ... def __init__( - self: MenuCommand[_AnyCallbackSigT,_MenuTypeT], + self: MenuCommand[_AnyCallbackSigT, _MenuTypeT], callback: _CallbackishT[_AnyCallbackSigT], type_: _MenuTypeT, name: str,