nemo_rl.models.generation.dynamo.arguments#

Deterministic argument and environment construction for managed Dynamo.

Module Contents#

Classes#

Functions#

_normalise_flag

_flag_for_key

_serialise_value

build_dynamo_vllm_argv

Build a conflict-free python -m dynamo.vllm argument list.

build_dynamo_frontend_argv

Build the managed dynamo.frontend argument list.

build_managed_worker_env

Return a reproducible worker environment with semantic overrides blocked.

redact_argv

Redact values following explicit credential options for safe logging.

redact_environment

Redact credential-like environment values before logging.

Data#

API#

nemo_rl.models.generation.dynamo.arguments._MANAGED_FLAGS#

None

nemo_rl.models.generation.dynamo.arguments._CREDENTIAL_FLAGS#

None

nemo_rl.models.generation.dynamo.arguments._normalise_flag(flag: str) str#
nemo_rl.models.generation.dynamo.arguments._flag_for_key(key: str) str#
nemo_rl.models.generation.dynamo.arguments._serialise_value(value: Any) str#
class nemo_rl.models.generation.dynamo.arguments._ArgvBuilder#

Initialization

add(flag: str, value: Any = None, *, source: str) None#
add_raw(args: collections.abc.Sequence[str], *, source: str) None#
nemo_rl.models.generation.dynamo.arguments.build_dynamo_vllm_argv(
*,
model_name: str,
namespace: str,
seed: int,
vllm_cfg: collections.abc.Mapping[str, Any],
vllm_kwargs: collections.abc.Mapping[str, Any],
dynamo_cfg: nemo_rl.models.generation.dynamo.config.DynamoCfg,
) list[str]#

Build a conflict-free python -m dynamo.vllm argument list.

nemo_rl.models.generation.dynamo.arguments.build_dynamo_frontend_argv(
*,
host: str,
port: int,
namespace: str,
dynamo_cfg: nemo_rl.models.generation.dynamo.config.DynamoCfg,
) list[str]#

Build the managed dynamo.frontend argument list.

nemo_rl.models.generation.dynamo.arguments.build_managed_worker_env(
*,
base_env: collections.abc.Mapping[str, str],
configured_env: collections.abc.Mapping[str, str],
manager_env: collections.abc.Mapping[str, str],
) dict[str, str]#

Return a reproducible worker environment with semantic overrides blocked.

nemo_rl.models.generation.dynamo.arguments.redact_argv(argv: collections.abc.Sequence[str]) list[str]#

Redact values following explicit credential options for safe logging.

nemo_rl.models.generation.dynamo.arguments.redact_environment(
env: collections.abc.Mapping[str, str],
) dict[str, str]#

Redact credential-like environment values before logging.