Fix types

synapse/util/caches/descriptors.py

@@ -21,6 +21,7 @@
     Awaitable,
     Callable,
     Collection,
+    Coroutine,
     Dict,
     Generic,
     Hashable,
@@ -58,7 +59,7 @@
 
 
 class CachedFunction(Generic[F]):
-    invalidate: Callable[[Tuple[Any, ...]], None]
+    invalidate: Callable[[Tuple[Any, ...]], Union[None, Coroutine]]
     invalidate_all: Callable[[], None]
     invalidate_external: Any = None
     prefill: Callable[[Tuple[Any, ...], Any], None]

synapse/util/caches/redis_caches.py

@@ -8,6 +8,7 @@
     Generic,
     List,
     Optional,
+    Tuple,
     Union,
     cast,
 )
@@ -75,7 +76,7 @@ async def _wrapped(self: Any, *args: Any, **kwargs: Any) -> Any:
                 await redis_shard_cache.set(cache_name, cache_key, value)
             return value
 
-        async def _invalidate(key: KT) -> None:
+        async def _invalidate(key: Tuple[Any, ...]) -> None:
             return await redis_shard_cache.delete(
                 cache_name,
                 _redis_key(key),

tests/handlers/test_register.py

@@ -138,7 +138,7 @@ class RegistrationTestCase(unittest.HomeserverTestCase):
     """Tests the RegistrationHandler."""
 
     def make_homeserver(self, reactor: MemoryReactor, clock: Clock) -> HomeServer:
-        reactor.advance(1)
+        reactor.advance(1)  # type: ignore
         hs_config = self.default_config()
 
         # some of the tests rely on us having a user consent version