nemo_gym.cli.setup_command

Module Contents

Functions

Name	Description
`_get_nemo_gym_install_flags`	Build uv pip install flags for nemo-gym in sub-venvs.
`_get_nemo_gym_version_spec`	Detect nemo-gym version from parent venv and return version specifier.
`get_venv_path`	Resolve the venv a server runs from: `uv_venv_dir/<type>/<name>/.venv`, else `<server dir>/.venv`.
`run_command`	-
`setup_env_command`	-

API

nemo_gym.cli.setup_command._get_nemo_gym_install_flags() -> str

Build uv pip install flags for nemo-gym in sub-venvs.

Supports:

Pre-release versions via NEMO_GYM_ALLOW_PRERELEASE=true
Custom PyPI indexes via UV_INDEX_URL, UV_EXTRA_INDEX_URL, UV_INDEX_STRATEGY
Auto-detection of parent venv version for consistency

Returns: str

String of flags to add to ‘uv pip install nemo-gym’

nemo_gym.cli.setup_command._get_nemo_gym_version_spec(
    is_editable_install: bool
) -> str

Detect nemo-gym version from parent venv and return version specifier.

Parameters:

is_editable_install

bool

Whether nemo-gym is installed in editable mode in parent venv

Returns: str

Version specifier string (e.g., “==0.2.1rc0”) or empty string

nemo_gym.cli.setup_command.get_venv_path(
    dir_path: pathlib.Path,
    global_config_dict: omegaconf.DictConfig
) -> pathlib.Path

Resolve the venv a server runs from: uv_venv_dir/<type>/<name>/.venv, else <server dir>/.venv.

Callers need this to launch a server with the venv’s own interpreter rather than trusting whatever bin/activate puts on PATH. A venv copied or moved after creation keeps the original prefix hard-coded in bin/activate, so sourcing it silently hands the server a different interpreter than the one uv_venv_dir asked for.

nemo_gym.cli.setup_command.run_command(
    command: str,
    working_dir_path: pathlib.Path,
    server_name: str = '',
    project_root: pathlib.Path | None = None,
    global_config_dict: omegaconf.DictConfig | None = None,
    stdout_target: typing.IO[typing.Any] | None = None,
    stderr_target: typing.IO[typing.Any] | None = None,
    extra_env: typing.Mapping[str, str] | None = None
) -> subprocess.Popen

nemo_gym.cli.setup_command.setup_env_command(
    dir_path: pathlib.Path,
    global_config_dict: omegaconf.DictConfig,
    prefix: str
) -> str