generator type

Fraunhofer-AISEC · Oct 30, 2024 · fb30be7 · fb30be7
1 parent e6c639f
commit fb30be7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...uage-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/ExpressionHandler.kt b/...uage-python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/ExpressionHandler.kt
@@ -119,6 +119,7 @@ class ExpressionHandler(frontend: PythonLanguageFrontend) :
         return newCollectionComprehension(rawNode = node).apply {
             statement = handle(node.elt)
             comprehensionExpressions += node.generators.map { handleComprehension(it) }
+            type = objectType("Generator")
         }
     }