async accepts `any`

1 files changed, 11 insertions, 4 deletions

diff --git a/script/core/diagnostics/not-yieldable.lua b/script/core/diagnostics/not-yieldable.lua
index 0be92e82..612b5b98 100644
--- a/script/core/diagnostics/not-yieldable.lua
+++ b/script/core/diagnostics/not-yieldable.lua
@@ -3,6 +3,7 @@ local await = require 'await'
 local guide = require 'parser.guide'
 local vm    = require 'vm'
 local lang  = require 'language'
+local infer = require 'core.infer'
 
 local function isYieldAble(defs, i)
     local hasFuncDef
@@ -10,15 +11,21 @@ local function isYieldAble(defs, i)
         if def.type == 'function' then
             hasFuncDef = true
             local arg = def.args and def.args[i]
-            if arg and vm.isAsync(arg, true) then
-                return true
+            if arg then
+                if infer.hasType(arg, 'any')
+                or vm.isAsync(arg, true) then
+                    return true
+                end
             end
         end
         if def.type == 'doc.type.function' then
             hasFuncDef = true
             local arg = def.args and def.args[i]
-            if arg and vm.isAsync(arg.extends, true) then
-                return true
+            if arg then
+                if infer.hasType(arg.extends, 'any')
+                or vm.isAsync(arg.extends, true) then
+                    return true
+                end
             end
         end
     end