Add additional typing.

matrix-org · Sep 2, 2020 · ca31cfe · ca31cfe
1 parent fa3cb7f
commit ca31cfe
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/synapse/storage/database.py b/synapse/storage/database.py
@@ -28,6 +28,7 @@
     Optional,
     Tuple,
     TypeVar,
+    cast,
     overload,
 )
 
@@ -507,8 +508,8 @@ def new_transaction(
             sql_txn_timer.labels(desc).observe(duration)
 
     async def runInteraction(
-        self, desc: str, func: Callable, *args: Any, **kwargs: Any
-    ) -> Any:
+        self, desc: str, func: "Callable[..., R]", *args: Any, **kwargs: Any
+    ) -> R:
         """Starts a transaction on the database and runs a given function
 
         Arguments:
@@ -547,7 +548,7 @@ async def runInteraction(
                 after_callback(*after_args, **after_kwargs)
             raise
 
-        return result
+        return cast(R, result)
 
     async def runWithConnection(
         self, func: "Callable[..., R]", *args: Any, **kwargs: Any