fix from review.

wemake-services · May 21, 2021 · 77c34db · 77c34db
1 parent 6893743
commit 77c34db
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/flake8_eradicate.py b/flake8_eradicate.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-
+import ast
 import tokenize
 from typing import Iterable, Tuple
 
@@ -25,7 +25,15 @@ class Checker(object):
 
     options = None
 
-    def __init__(self, tree, filename: str):    # noqa: D107
+    def __init__(self, tree: ast.AST, filename: str):
+        """
+        ``flake8`` plugin constructor.
+
+        Arguments:
+            tree: the file abstract syntax tree.
+            filename: the name of the file to process
+
+        """
         self.filename = filename
 
         self._options = {

diff --git a/tests/test_comments.py b/tests/test_comments.py
@@ -136,12 +136,12 @@ def test_lines_with_commented_out_code_incorrect_fixture_output(absolute_path):
 
     OptionsStub = namedtuple(
         'Options',
-        'eradicate_aggressive eradicate_whitelist eradicate_whitelist_extend'
+        'eradicate_aggressive eradicate_whitelist eradicate_whitelist_extend',
     )
     Checker.options = OptionsStub(
         eradicate_aggressive=True,
         eradicate_whitelist=False,
-        eradicate_whitelist_extend=False
+        eradicate_whitelist_extend=False,
     )
 
     checker = Checker(tree=None, filename=filename)