Add simple definition of the return value of check_git_attr in pyi.

Kentzo · Feb 9, 2020 · 70a73bf · 70a73bf
1 parent 1443832
commit 70a73bf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/git_archive_all.pyi b/git_archive_all.pyi
@@ -4,6 +4,7 @@ from typing import Callable, Collection, ClassVar, Dict, Generator, Iterable, Li
 
 PathLike = Union[str, bytes, _PathLike]
 PathStr = Union[str, bytes]
+CheckGitAttrGen = Generator[Dict[str, bytes], PathStr, None]
 
 def fsdecode(filename: PathLike) -> str: ...
 
@@ -23,7 +24,7 @@ class GitArchiver(object):
     LOG: ClassVar[logging.Logger]
 
     _should_decode_path: Optional[bool]
-    _check_attr_gens: Dict[str, Generator[Dict[str, bytes], PathStr, None]]
+    _check_attr_gens: Dict[str, CheckGitAttrGen]
     git_version: Optional[Tuple[int]]
     main_repo_abspath: PathStr
     prefix: PathStr
@@ -51,7 +52,7 @@ class GitArchiver(object):
 
     def walk_git_files(self, repo_path: PathStr = None) -> Generator[PathStr, None, None]: ...
 
-    def check_git_attr(self, repo_abspath: PathStr, attrs: Collection[str]) -> Generator[Dict[str, bytes], PathStr, None]: ...
+    def check_git_attr(self, repo_abspath: PathStr, attrs: Collection[str]) -> CheckGitAttrGen: ...
 
     def resolve_git_main_repo_abspath(self, abspath: PathLike) -> PathStr: ...