fix getting closure for GeneratorFunctionDefinitionNode

cosminbasca · cosminbasca · commit a27d4896adc3 · 2019-08-14T14:49:24.000+02:00
diff --git a/graalpython/com.oracle.graal.python/src/com/oracle/graal/python/nodes/function/GeneratorFunctionDefinitionNode.java b/graalpython/com.oracle.graal.python/src/com/oracle/graal/python/nodes/function/GeneratorFunctionDefinitionNode.java
@@ -86,7 +86,7 @@ public PGeneratorFunction execute(VirtualFrame frame) {
                 kwDefaultValues[i] = new PKeyword(kwDefaults[i].name, kwDefaults[i].execute(frame));
             }
         }
-        PCell[] closure = getClosureFromLocals(frame);
+        PCell[] closure = getClosureFromGeneratorOrFunctionLocals(frame);
         return withDocString(frame, factory().createGeneratorFunction(functionName, enclosingClassName, getGeneratorCallTarget(), PArguments.getGlobals(frame), closure, defaultValues,
                         kwDefaultValues));
     }

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,7 @@ public PGeneratorFunction execute(VirtualFrame frame) {`
`86`	`86`	`kwDefaultValues[i] = new PKeyword(kwDefaults[i].name, kwDefaults[i].execute(frame));`
`87`	`87`	`}`
`88`	`88`	`}`
`89`		`- PCell[] closure = getClosureFromLocals(frame);`
	`89`	`+ PCell[] closure = getClosureFromGeneratorOrFunctionLocals(frame);`
`90`	`90`	`return withDocString(frame, factory().createGeneratorFunction(functionName, enclosingClassName, getGeneratorCallTarget(), PArguments.getGlobals(frame), closure, defaultValues,`
`91`	`91`	`kwDefaultValues));`
`92`	`92`	`}`