Fix

cqfn · Jul 8, 2020 · 3462bc4 · 3462bc4
1 parent b0275a9
commit 3462bc4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/aibolit/__main__.py b/aibolit/__main__.py
@@ -38,6 +38,8 @@
 from pathlib import Path
 from sys import stdout
 from typing import List
+
+import javalang
 import numpy as np  # type: ignore
 import requests
 from lxml import etree  # type: ignore
@@ -163,7 +165,7 @@ def flatten(l):
     return [item for sublist in l for item in sublist]
 
 
-def find_annotation_by_node_type(tree, node_type):
+def find_annotation_by_node_type(tree: javalang.tree.CompilationUnit, node_type: javalang.tree.Node):
     """Search nodes with annotations.
 
     :param tree: javalang.tree