remove optionals for all vars of for .. in

Author: sumneko

script/vm/compiler.lua

@@ -911,12 +911,8 @@ local function compileForVars(source)
     if source.keys then
         for i, loc in ipairs(source.keys) do
             local node = getReturn(source._iterator, i, source._iterArgs)
-            if node then
-                if i == 1 then
-                    node:removeOptional()
-                end
-                source._iterVars[loc] = node
-            end
+            node:removeOptional()
+            source._iterVars[loc] = node
         end
     end
 end

test/type_inference/init.lua

@@ -3234,3 +3234,19 @@ local <?x?> = f()
 TEST 'number' [=[
 local <?x?> = X --[[@as number]]
 ]=]
+
+TEST 'number' [[
+---@return number?, number?
+local function f() end
+
+for <?x?>, y in f do
+end
+]]
+
+TEST 'number' [[
+---@return number?, number?
+local function f() end
+
+for x, <?y?> in f do
+end
+]]