Merge branch 'number_of_methods' of https://github.com/lukyanoffpasho…

…k/aibolit into number_of_methods
cqfn · Aug 13, 2020 · 3a9bd8e · 3a9bd8e
2 parents ee7f7bf + 2a808bb
commit 3a9bd8e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/aibolit/ast_framework/computed_fields_catalog/nodes_filter.py b/aibolit/ast_framework/computed_fields_catalog/nodes_filter.py
@@ -35,11 +35,9 @@ def nodes_filter_factory(
 
     def filter(base_node: ASTNode) -> Iterator[ASTNode]:
         base_field = getattr(base_node, base_field_name)
-        if isinstance(base_field, list) and all(
-            isinstance(item, ASTNode) for item in base_field
-        ):
+        if isinstance(base_field, list):
             for node in base_field:
-                if node.node_type in node_types:
+                if isinstance(node, ASTNode) and node.node_type in node_types:
                     yield node
         else:
             raise RuntimeError(