From e820d8c1d50f3d9b29c00182274cd38d93aa5fb0 Mon Sep 17 00:00:00 2001 From: lladhibhutall Date: Tue, 7 Apr 2020 23:20:03 +0530 Subject: [PATCH 1/2] Add overloaded methods for return type of None --- stdlib/3/typing.pyi | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/stdlib/3/typing.pyi b/stdlib/3/typing.pyi index 11e7ed76a3f8..6ef9d5bcf906 100644 --- a/stdlib/3/typing.pyi +++ b/stdlib/3/typing.pyi @@ -548,7 +548,11 @@ class Match(Generic[AnyStr]): # TODO: The return for a group may be None, except if __group is 0 or not given. @overload - def group(self, __group: Union[str, int] = ...) -> AnyStr: ... + def group(self, __group: Literal[None]= ...) -> AnyStr: ... + @overload + def group(self, __group: Literal[0]= ...) -> AnyStr: ... + @overload + def group(self, __group: Union[str, int] = ...) -> Optional[AnyStr]: ... @overload def group( self, From ab6f5eec45303f1900227dd5ca15c7fe934c64fa Mon Sep 17 00:00:00 2001 From: lladhibhutall Date: Tue, 7 Apr 2020 23:28:13 +0530 Subject: [PATCH 2/2] Flake8 --- stdlib/3/typing.pyi | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stdlib/3/typing.pyi b/stdlib/3/typing.pyi index 6ef9d5bcf906..6508d09abe2f 100644 --- a/stdlib/3/typing.pyi +++ b/stdlib/3/typing.pyi @@ -548,9 +548,9 @@ class Match(Generic[AnyStr]): # TODO: The return for a group may be None, except if __group is 0 or not given. @overload - def group(self, __group: Literal[None]= ...) -> AnyStr: ... + def group(self, __group: Literal[None] = ...) -> AnyStr: ... @overload - def group(self, __group: Literal[0]= ...) -> AnyStr: ... + def group(self, __group: Literal[0] = ...) -> AnyStr: ... @overload def group(self, __group: Union[str, int] = ...) -> Optional[AnyStr]: ... @overload