Use the await method properly and just use it instead of some crazy things

2018-02-21 00:27:15 +01:00
parent bf01b9d47c
commit bc0210af70
3 changed files with 8 additions and 11 deletions
--- a/jedi/evaluate/context/asynchronous.py
+++ b/jedi/evaluate/context/asynchronous.py
@@ -18,7 +18,8 @@ class AsyncBase(BuiltinOverwrite):
 class Coroutine(AsyncBase):
    special_object_identifier = u'COROUTINE_TYPE'

-    def execute_await(self):
+    @publish_method('__await__')
+    def _await(self):
        return self._func_execution_context.get_return_values()