update all mains

Author: DNXie

.meta/mast/main.py

@@ -15,7 +15,7 @@
     MastLauncher,
     mount_mnt_directory,
 )
-from forge.controller.provisioner import init_provisioner
+from forge.controller.provisioner import get_or_create_provisioner
 
 from forge.types import (
     Launcher,
@@ -68,7 +68,9 @@ async def main(cfg: DictConfig, mode: str = "detached", extra_args: list = None)
     else:
         # In remote mode, we're already running inside MAST, so mount directory, init provisioner and run training
         mount_mnt_directory("/mnt/wsfuse")
-        await init_provisioner(ProvisionerConfig(launcher_config=launcher_config))
+        await get_or_create_provisioner(
+            ProvisionerConfig(launcher_config=launcher_config)
+        )
         await grpo_main(cfg)
 
 

apps/grpo/main.py

@@ -25,7 +25,7 @@
 from forge.actors.replay_buffer import ReplayBuffer
 from forge.actors.trainer import RLTrainer
 from forge.controller.actor import ForgeActor
-from forge.controller.provisioner import init_provisioner, shutdown
+from forge.controller.provisioner import get_or_create_provisioner, shutdown
 from forge.data.rewards import MathReward, ThinkingReward
 from forge.data_models.completion import Completion
 from forge.observability.metric_actors import get_or_create_metric_logger
@@ -298,11 +298,11 @@ async def main(cfg: DictConfig):
     # ---- Global setups ---- #
     provisioner = None
     if cfg.get("provisioner", None) is not None:
-        provisioner = await init_provisioner(
+        provisioner = await get_or_create_provisioner(
             ProvisionerConfig(launcher_config=LauncherConfig(**cfg.provisioner))
         )
     else:
-        provisioner = await init_provisioner()
+        provisioner = await get_or_create_provisioner()
 
     metric_logging_cfg = cfg.get("metric_logging", {"console": {"log_per_rank": False}})
     mlogger = await get_or_create_metric_logger(process_name="Controller")