Fix type for tree argument (#353)

Author: adamchainz

HISTORY.rst

@@ -2,6 +2,11 @@
 History
 =======
 
+* Fix type hint for ``tree`` argument.
+
+  Thanks to kasium for the report in `Issue #352
+  <https://github.com/adamchainz/flake8-comprehensions/issues/352>`__.
+
 3.6.0 (2021-08-13)
 ------------------
 

src/flake8_comprehensions/__init__.py

@@ -18,7 +18,7 @@ class ComprehensionChecker:
 
     __slots__ = ("tree",)
 
-    def __init__(self, tree: ast.Module) -> None:
+    def __init__(self, tree: ast.AST) -> None:
         self.tree = tree
 
     messages = {