Skip to content

Commit a1e185f

Browse files
Update forge references to torchforge in MAST setup (#615)
1 parent 0754eff commit a1e185f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.meta/mast/client_bootstrap.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ if [ -z "$WORKSPACE_DIR" ] || [ ! -d "$WORKSPACE_DIR" ]; then
3939
WORKSPACE_DIR="$CONDA_PREFIX"
4040
fi
4141

42-
cd "$WORKSPACE_DIR/forge"
42+
cd "$WORKSPACE_DIR/torchforge"
4343

4444
# Execute the client training script with all passed arguments
4545
exec python -X faulthandler .meta/mast/main.py "$@"

src/forge/controller/launcher.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def __init__(
206206
self.timeout_sec = 1 * 60 * 60 # Kill the job if idle for 1 hour
207207
self.user = getpass.getuser()
208208
self.work_dir = f"/home/{self.user}"
209-
self.edittable_workspaces = ["forge"]
209+
self.edittable_workspaces = ["torchforge"]
210210
self.remote_work_dir = "/packages/monarch_default_workspace/workspace/"
211211
self.editable_workspace_paths = [
212212
f"{self.work_dir}/{workspace}" for workspace in self.edittable_workspaces
@@ -367,7 +367,7 @@ def _create_client_role(self, appdef: specs.AppDef) -> specs.Role:
367367
# Override with client-specific configuration
368368
client_role.name = "client"
369369
# Use the bootstrap script as entrypoint
370-
client_role.entrypoint = "workspace/forge/.meta/mast/client_bootstrap.sh"
370+
client_role.entrypoint = "workspace/torchforge/.meta/mast/client_bootstrap.sh"
371371

372372
# Build args for the client role (passed to the bootstrap script)
373373
# These args will be passed to client_bootstrap.sh which forwards them to main.py

0 commit comments

Comments
 (0)