Lint

Author: domdfcoding

flake8_dunder_all/__init__.py

@@ -32,7 +32,7 @@
 # stdlib
 import ast
 import sys
-from typing import Any, Generator, Set, Tuple, Type, Union
+from typing import Any, Generator, List, Set, Tuple, Type, Union
 
 # 3rd party
 from consolekit.terminal_colours import Fore
@@ -236,7 +236,7 @@ def check_and_add_all(filename: PathLike, quote_type: str = '"') -> int:
 
 	.. versionchanged:: 0.2.0
 
-		Now returns ``0`` and doesn't add ``__all__`` if the file contains a ``# noqa: DALL000`` comment.
+		Now returns ``0`` and doesn't add ``__all__`` if the file contains a ``noqa: DALL000`` comment.
 	"""
 
 	quotes = {"'", '"'}
@@ -250,7 +250,8 @@ def check_and_add_all(filename: PathLike, quote_type: str = '"') -> int:
 		for line in source.splitlines():
 			noqas = find_noqa(line)
 			if noqas is not None:
-				if "DALL000" in noqas.group(1).rstrip().upper().split(','):
+				noqa_list: List[str] = noqas.group(1).rstrip().upper().split(',')
+				if "DALL000" in noqa_list:
 					return 0
 
 		tree = ast.parse(source)