From 431e3e3ed92712e4c53dddd30a3d8149d238da38 Mon Sep 17 00:00:00 2001
From: Andrew Morgan <andrew@amorgan.xyz>
Date: Fri, 11 Sep 2020 15:58:00 +0100
Subject: [PATCH] Replace verbose type with StateMap

---
 synapse/events/third_party_rules.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/synapse/events/third_party_rules.py b/synapse/events/third_party_rules.py
index 81e3c4eb8a..af5032637a 100644
--- a/synapse/events/third_party_rules.py
+++ b/synapse/events/third_party_rules.py
@@ -12,11 +12,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import Dict, Tuple
-
 from synapse.events import EventBase
 from synapse.events.snapshot import EventContext
-from synapse.types import Requester
+from synapse.types import Requester, StateMap
 
 
 class PublicRoomsManager:
@@ -170,7 +168,7 @@ class ThirdPartyEventRules:
 
     async def _get_state_events_dict_for_room(
         self, room_id: str
-    ) -> Dict[Tuple[str, str], EventBase]:
+    ) -> StateMap[EventBase]:
         """Given a room ID, return the state events of that room.
 
         Args: