diff --git a/python/monarch/_src/actor/actor_mesh.py b/python/monarch/_src/actor/actor_mesh.py index fe744ee96..43aa91758 100644 --- a/python/monarch/_src/actor/actor_mesh.py +++ b/python/monarch/_src/actor/actor_mesh.py @@ -392,12 +392,15 @@ def enable_transport(transport: "ChannelTransport | str") -> None: Currently only one transport type may be enabled at one time. In the future we may allow multiple to be enabled. + + For Meta usage, use metatls-hostname """ if isinstance(transport, str): transport = { "tcp": ChannelTransport.TcpWithHostname, "ipc": ChannelTransport.Unix, "metatls": ChannelTransport.MetaTlsWithIpV6, + "metatls-hostname": ChannelTransport.MetaTlsWithHostname, }.get(transport) if transport is None: raise ValueError(f"unknown transport: {transport}") diff --git a/python/monarch/_src/job/meta.py b/python/monarch/_src/job/meta.py index 0adcdcef5..f7460f812 100644 --- a/python/monarch/_src/job/meta.py +++ b/python/monarch/_src/job/meta.py @@ -22,13 +22,12 @@ from monarch._src.actor.allocator import AllocateMixin from monarch._src.actor.host_mesh import host_mesh_from_alloc from monarch._src.actor.meta.allocator import ( - MastAllocatorBase, MastAllocatorConfig, MastHostAllocator, MastHostAllocatorBase, ) -from monarch._src.job.job import BatchJob, enable_transport, JobState, JobTrait +from monarch._src.job.job import BatchJob, JobState, JobTrait from monarch.tools.commands import create, info, kill from monarch.tools.components.meta import hyperactor @@ -40,8 +39,6 @@ from torchx.specs import AppState from torchx.specs.fb.component_helpers import Packages -enable_transport("metatls") - class _MASTSpec(NamedTuple): hpcIdentity: str