diff --git a/tractor/__init__.py b/tractor/__init__.py index 9ff62cc..a7cadb9 100644 --- a/tractor/__init__.py +++ b/tractor/__init__.py @@ -15,7 +15,11 @@ from ._streaming import ( from ._discovery import get_arbiter, find_actor, wait_for_actor from ._trionics import open_nursery from ._state import current_actor, is_root_process -from ._exceptions import RemoteActorError, ModuleNotExposed +from ._exceptions import ( + RemoteActorError, + ModuleNotExposed, + ContextCancelled, +) from ._debug import breakpoint, post_mortem from . import msg from ._root import run, run_daemon, open_root_actor @@ -27,6 +31,7 @@ __all__ = [ 'ModuleNotExposed', 'MultiError', 'RemoteActorError', + 'ContextCancelled', 'breakpoint', 'current_actor', 'find_actor', @@ -40,6 +45,8 @@ __all__ = [ 'run_daemon', 'stream', 'context', + 'ReceiveMsgStream', + 'MsgStream', 'to_asyncio', 'wait_for_actor', ]