nemo_gym.orchestration.executors.script_templates

Module Contents

Functions

Name	Description
`bash_var`	-
`render_driver_entrypoint`	Render the srun entrypoint for the driver step.
`render_gym_cmd`	Render a bash array with each argument on its own line.
`render_health_check`	-

Data

_HEALTH_WAIT_MULTI

API

nemo_gym.orchestration.executors.script_templates.bash_var(
    name: str
) -> str

nemo_gym.orchestration.executors.script_templates.render_driver_entrypoint(
    repo: str | None,
    ref: str | None,
    prepare_cmd: str | None
) -> str

Render the srun entrypoint for the driver step.

When either gym_install or prepare is needed, wraps everything in a single bash -c so prepare and run happen in the same srun step and container.

nemo_gym.orchestration.executors.script_templates.render_gym_cmd(
    subcommand: str,
    var_name: str,
    args: list[str]
) -> str

Render a bash array with each argument on its own line.

nemo_gym.orchestration.executors.script_templates.render_health_check(
    name: str,
    port: int,
    path: str,
    timeout: int
) -> str

nemo_gym.orchestration.executors.script_templates._HEALTH_WAIT_MULTI = '# Wait for {name} (try multiple health endpoints)\necho "Waiting for {name} at ...