[#570] Relax nodes_filter restriction on base fields.

cqfn · Aug 6, 2020 · 2a808bb · 2a808bb
1 parent 587ffee
commit 2a808bb
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(