diff --git a/aiofiles/os.py b/aiofiles/os.py index 9b5c3ea..b9fe473 100644 --- a/aiofiles/os.py +++ b/aiofiles/os.py @@ -5,13 +5,12 @@ def wrap(func): - @asyncio.coroutine @wraps(func) - def run(*args, loop=None, executor=None, **kwargs): + async def run(*args, loop=None, executor=None, **kwargs): if loop is None: loop = asyncio.get_event_loop() pfunc = partial(func, *args, **kwargs) - return loop.run_in_executor(executor, pfunc) + return await loop.run_in_executor(executor, pfunc) return run