tractor/__init__.py

@@ -67,4 +67,4 @@ from ._root import (
 from ._ipc import Channel as Channel
 from ._portal import Portal as Portal
 from ._runtime import Actor as Actor
-# from . import hilevel as hilevel
+from . import hilevel as hilevel

tractor/_context.py

@@ -47,9 +47,6 @@ from functools import partial
 import inspect
 from pprint import pformat
 import textwrap
-from types import (
-    UnionType,
-)
 from typing import (
     Any,
     AsyncGenerator,
@@ -2551,14 +2548,7 @@ def context(
     name: str
     param: Type
     for name, param in annots.items():
-        if (
+        if param is Context:
-            param is Context
-            or (
-                isinstance(param, UnionType)
-                and
-                Context in param.__args__
-            )
-        ):
             ctx_var_name: str = name
             break
     else: