Update type annotations.

Author: Loren Arthur

tornado/gen.py

@@ -700,13 +700,13 @@ async def run(self) -> None:
             future = self.future
             if future is None:
                 raise Exception("No pending future")
-            if not self.future.done():
+            if not self.future.done(): # type: ignore
                 _step = asyncio.Event()
 
-                def step(*args):
+                def step(f: "Future[_T]") -> None:
                     _step.set()
 
-                self.io_loop.add_future(self.future, step)
+                self.io_loop.add_future(self.future, step) # type: ignore
                 await _step.wait()
             self.future = None
             try:
@@ -748,18 +748,17 @@ def step(*args):
             if self.future is moment:
                 await sleep(0)
 
-    def handle_yield(self, yielded: _Yieldable) -> bool:
+    def handle_yield(self, yielded: _Yieldable) -> None:
         try:
             self.future = convert_yielded(yielded)
         except BadYieldError:
             self.future = Future()
             future_set_exc_info(self.future, sys.exc_info())
 
-    def finish(self, future):
+    def finish(self, future: "Future[_T]") -> None:
         if future.cancelled():
             self.task.cancel()
-            self.future.cancel()
-        self.task = None
+            self.future.cancel() # type: ignore
 
 
 # Convert Awaitables into Futures.

tornado/ioloop.py

@@ -506,9 +506,11 @@ def run() -> None:
                     future_cell[0] = fut
                     fut.set_result(result)
             assert future_cell[0] is not None
-            self.add_future(
-                future_cell[0], lambda f: f.add_done_callback(lambda _: self.stop())
-            )
+
+            def _stop(f: "Future[_T]") -> None:
+                f.add_done_callback(lambda _: self.stop())
+
+            self.add_future(future_cell[0], _stop)
 
         self.add_callback(run)
         if timeout is not None: