fixes #808

Author: jph00

fastcore/script.py

@@ -7,7 +7,7 @@
            'set_ctx', 'call_parse']
 
 # %% ../nbs/06_script.ipynb #8a36db98
-import inspect,argparse,shutil,types
+import inspect,argparse,shutil,types,asyncio
 
 from functools import wraps,partial
 from .imports import *
@@ -167,7 +167,8 @@ def _f(*args, **kwargs):
             args = args.__dict__
             xtra = otherwise(args.pop('xtra', ''), eq(1), p.prog)
             tfunc = trace(func) if args.pop('pdb', False) else func
-            return tfunc(**merge(args, args_from_prog(func, xtra)))
+            res = tfunc(**merge(args, args_from_prog(func, xtra)))
+            return asyncio.run(res) if inspect.isawaitable(res) else res
 
     mod = inspect.getmodule(inspect.currentframe().f_back)
     if getattr(mod, '__name__', '') =="__main__":

nbs/06_script.ipynb

@@ -178,7 +178,7 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "import inspect,argparse,shutil,types\n",
+    "import inspect,argparse,shutil,types,asyncio\n",
     "\n",
     "from functools import wraps,partial\n",
     "from fastcore.imports import *\n",
@@ -804,7 +804,8 @@
     "            args = args.__dict__\n",
     "            xtra = otherwise(args.pop('xtra', ''), eq(1), p.prog)\n",
     "            tfunc = trace(func) if args.pop('pdb', False) else func\n",
-    "            return tfunc(**merge(args, args_from_prog(func, xtra)))\n",
+    "            res = tfunc(**merge(args, args_from_prog(func, xtra)))\n",
+    "            return asyncio.run(res) if inspect.isawaitable(res) else res\n",
     "\n",
     "    mod = inspect.getmodule(inspect.currentframe().f_back)\n",
     "    if getattr(mod, '__name__', '') ==\"__main__\":\n",