Skip to content

Commit 04a1ebc

Browse files
committed
fix remote_setup for MAST
1 parent 2069b3d commit 04a1ebc

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/forge/controller/launcher.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ async def initialize(self) -> None:
114114
async def get_allocator(self, name: str, num_hosts: int) -> tuple[Any, Any, str]:
115115
pass
116116

117-
async def remote_setup(self, procs: ProcMesh) -> tuple[str, int]:
117+
async def remote_setup(self, procs: ProcMesh) -> None:
118118
pass
119119

120120

@@ -156,7 +156,7 @@ async def get_allocator(self, name: str, num_hosts: int) -> tuple[Any, Any, str]
156156
server_name = f"slurm:///{server_info.name}"
157157
return alloc, None, server_name # (Allocator, AllocConstraints, SeverName)
158158

159-
async def remote_setup(self, procs: ProcMesh) -> tuple[str, int]:
159+
async def remote_setup(self, procs: ProcMesh) -> None:
160160
return
161161

162162

@@ -201,10 +201,9 @@ async def get_allocator(self, name: str, num_hosts: int) -> tuple[Any, Any, str]
201201

202202
return allocator, alloc_constraints, self.create_server_handle()
203203

204-
async def remote_setup(self, procs: ProcMesh) -> tuple[str, int]:
204+
async def remote_setup(self, procs: ProcMesh) -> None:
205205
setup = procs.spawn(f"setup-{uuid.uuid1()}", MastSetupActor)
206206
await setup.mount.call(mount_dst="/mnt/wsfuse")
207-
return await setup.get_info.choose()
208207

209208
async def launch_mast_job(self):
210209
handle = self.create_server_handle()

0 commit comments

Comments
 (0)