From c58a6ea80f6a8af6ffc14d9d29e506dfa351547a Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <tgoodlet@gmail.com>
Date: Wed, 16 Jan 2019 16:50:30 -0500
Subject: [PATCH] Fix type annots

---
 tractor/msg.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/tractor/msg.py b/tractor/msg.py
index 16c3982..52d240e 100644
--- a/tractor/msg.py
+++ b/tractor/msg.py
@@ -2,13 +2,12 @@
 Messaging pattern APIs and helpers.
 """
 import typing
-from typing import Dict
+from typing import Dict, Any
 from functools import partial
 
 import trio
 import wrapt
 
-from ._ipc import Context
 from .log import get_logger
 from . import current_actor
 
@@ -18,8 +17,8 @@ log = get_logger('messaging')
 
 
 async def fan_out_to_ctxs(
-    pub_gen: typing.AsyncGenerator,
-    topics2ctxs: Dict[str, Context],
+    pub_gen: typing.Callable,  # it's an async gen ... gd mypy
+    topics2ctxs: Dict[str, set],
     topic_key: str = 'key',
 ) -> None:
     """Request and fan out quotes to each subscribed actor channel.
@@ -30,7 +29,7 @@ async def fan_out_to_ctxs(
     async for published in pub_gen(
         get_topics=get_topics,
     ):
-        ctx_payloads = {}
+        ctx_payloads: Dict[str, Any] = {}
         for packet_key, data in published.items():
             # grab each suscription topic using provided key for lookup
             topic = data[topic_key]