diff --git a/src/forge/controller/launcher.py b/src/forge/controller/launcher.py index a11ab50be..b9e3336ea 100644 --- a/src/forge/controller/launcher.py +++ b/src/forge/controller/launcher.py @@ -15,13 +15,9 @@ from typing import Any import monarch - import torchx.specs as specs - -from forge.types import Launcher, LauncherConfig from monarch._rust_bindings.monarch_hyperactor.alloc import AllocConstraints from monarch._rust_bindings.monarch_hyperactor.channel import ChannelTransport - from monarch._rust_bindings.monarch_hyperactor.config import configure from monarch._src.actor.allocator import RemoteAllocator, TorchXRemoteAllocInitializer from monarch.actor import Actor, endpoint, ProcMesh @@ -29,6 +25,8 @@ from monarch.tools.commands import create, info from monarch.tools.config import Config, Workspace +from forge.types import Launcher, LauncherConfig + _MAST_AVAILABLE = False try: @@ -267,7 +265,7 @@ async def launch_mast_job(self): def add_additional_packages(self, packages: "Packages") -> "Packages": packages.add_package("oil.oilfs:stable") - packages.add_package("manifold.manifoldfs") + packages.add_package("manifold.manifoldfs:prod") return packages def build_appdef(self) -> specs.AppDef: