diff --git a/python/monarch/actor/__init__.py b/python/monarch/actor/__init__.py index 37b633ff9..b94d9ca94 100644 --- a/python/monarch/actor/__init__.py +++ b/python/monarch/actor/__init__.py @@ -18,6 +18,7 @@ as_endpoint, Channel, context, + Context, current_actor_name, current_rank, current_size, @@ -80,4 +81,5 @@ "run_worker_loop_forever", "attach_to_workers", "enable_transport", + "Context", ]