Accept the miles-train command where the fault injector needs the synchronous driver

The step-boundary fault actions recognized the synchronous driver only as
train.py, so a run launched with the installed miles-train command aborted.
The deployment-entrypoint test now reads the drivers' code where it lives,
in miles.entrypoints, including the main() the root scripts call.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

Author: HJSang

miles/utils/test_utils/fault_injector/actions/base.py

@@ -39,10 +39,12 @@ async def __call__(self, *, context: FaultHookContext, resources: FaultHookResou
 
 
 PARKABLE_TRAIN_SCRIPT: str = "train.py"
+# The same driver installed as a command (setup.py's console_scripts).
+PARKABLE_TRAIN_COMMAND: str = "miles-train"
 
 
 def _assert_synchronous_orchestrator(args: object, trainer_model_id: str | None, action: str) -> None:
-    assert (script := Path(sys.argv[0]).name) == PARKABLE_TRAIN_SCRIPT, (
+    assert (script := Path(sys.argv[0]).name) in (PARKABLE_TRAIN_SCRIPT, PARKABLE_TRAIN_COMMAND), (
         f"{action} runs between synchronous steps, and only {PARKABLE_TRAIN_SCRIPT} stands still there; "
         f"{script} has already started the next rollout by the time it reaches here"
     )

tests/fast/utils/test_utils/fault_injector/test_frozen.py

@@ -12,6 +12,7 @@
     PARKABLE_TRAIN_SCRIPT,
     FaultHookContext,
     FaultHookResources,
+    _assert_synchronous_orchestrator,
 )
 from miles.utils.test_utils.fault_injector.actions.cell import RestartRolloutCellAction
 from miles.utils.test_utils.fault_injector.actions.frozen import (
@@ -166,6 +167,15 @@ def test_the_parkable_script_is_the_default_launch_script(self) -> None:
         """The freeze guard and launch harness must agree on the synchronous script."""
         assert PARKABLE_TRAIN_SCRIPT == DEFAULT_TRAIN_SCRIPT
 
+    @pytest.mark.parametrize("script", ["/miles/train.py", "/venv/bin/miles-train"])
+    async def test_the_synchronous_driver_passes_however_it_was_launched(
+        self, monkeypatch: pytest.MonkeyPatch, tmp_path: Path, script: str
+    ) -> None:
+        """The checkout script and the installed command run the same synchronous loop."""
+        monkeypatch.setattr(sys, "argv", [script])
+        args = SimpleNamespace(ci_fault_hooks_path=str(tmp_path / "plan.json"), update_weights_interval=1)
+        _assert_synchronous_orchestrator(args, trainer_model_id=None, action="sleep_forever")
+
     async def test_an_unarmed_async_loop_is_not_subject_to_freeze_guards(
         self, configure_hooks: Callable[..., _FaultHookController], monkeypatch: pytest.MonkeyPatch
     ) -> None:

tests/fast/utils/workers/test_deployment_entrypoint.py

@@ -122,9 +122,10 @@ def test_a_deployment_that_carries_no_trainer_starts_nothing(self, monkeypatch):
         assert fake.calls == []
 
 
+# where each driver's code lives; train.py and train_async.py at the root only call these mains
 _DRIVER_SCRIPTS = [
-    "train.py",
-    "train_async.py",
+    "miles/entrypoints/train.py",
+    "miles/entrypoints/train_async.py",
     "train_multi_policy.py",
 ]
 
@@ -138,11 +139,12 @@ def test_a_driver_starts_its_training_without_asking_what_this_launch_deploys(se
 
 
 def _functions_called_in_main(script: str) -> set[str]:
+    """Calls in the script's ``if __name__ == "__main__"`` block, or in the ``main()`` that block runs."""
     tree = ast.parse((REPO_ROOT / script).read_text())
     return {
         ast.unparse(node.func)
         for statement in tree.body
-        if isinstance(statement, ast.If)
+        if isinstance(statement, ast.If) or (isinstance(statement, ast.FunctionDef) and statement.name == "main")
         for node in ast.walk(statement)
         if isinstance(node, ast.Call)
     }