Improve a type hint while I'm here

matrix-org · Jan 5, 2022 · 0999ed8 · 0999ed8
1 parent 9953d6c
commit 0999ed8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_utils/__init__.py b/tests/test_utils/__init__.py
@@ -19,7 +19,7 @@
 import warnings
 from asyncio import Future
 from binascii import unhexlify
-from typing import Any, Awaitable, Callable, TypeVar
+from typing import Awaitable, Callable, TypeVar
 from unittest.mock import Mock
 
 import attr
@@ -46,7 +46,7 @@ def get_awaitable_result(awaitable: Awaitable[TV]) -> TV:
     raise Exception("awaitable has not yet completed")
 
 
-def make_awaitable(result: Any) -> Awaitable[Any]:
+def make_awaitable(result: TV) -> Awaitable[TV]:
     """
     Makes an awaitable, suitable for mocking an `async` function.
     This uses Futures as they can be awaited multiple times so can be returned