API ReferenceFull Library ReferenceNemo GymNemo Gym

nemo_gym.cli_setup_command

View as Markdown

Module Contents

Functions

NameDescription
_get_nemo_gym_install_flagsBuild uv pip install flags for nemo-gym in sub-venvs.
_get_nemo_gym_version_specDetect nemo-gym version from parent venv and return version specifier.
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.run_command(
    command: str,
    working_dir_path: pathlib.Path,
    server_name: str = ''
) -> subprocess.Popen
 
nemo_gym.cli_setup_command.setup_env_command(
    dir_path: pathlib.Path,
    global_config_dict: omegaconf.DictConfig,
    prefix: str
) -> str