From 14114547e8f01253cbbb8e8638b74fc796b6ca66 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Sun, 2 May 2021 14:12:35 -0400
Subject: [PATCH] Expose `@context` decorator at top level

---
 tractor/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tractor/__init__.py b/tractor/__init__.py
index 7e6f800..25b56b1 100644
--- a/tractor/__init__.py
+++ b/tractor/__init__.py
@@ -5,7 +5,7 @@ tractor: An actor model micro-framework built on
 from trio import MultiError
 
 from ._ipc import Channel
-from ._streaming import Context, stream
+from ._streaming import Context, stream, context
 from ._discovery import get_arbiter, find_actor, wait_for_actor
 from ._trionics import open_nursery
 from ._state import current_actor, is_root_process
@@ -33,7 +33,7 @@ __all__ = [
     'run',
     'run_daemon',
     'stream',
-    'wait_for_actor',
+    'context',
     'to_asyncio',
     'wait_for_actor',
 ]