nemo_rl.utils.venvs

Contenu du module

Fonctions

Données

dir_path git_root DEFAULT_VENV_DIR logger

API

nemo_rl.utils.venvs.dir_path

Valeur: dirname(...)

nemo_rl.utils.venvs.git_root

Valeur: abspath(...)

nemo_rl.utils.venvs.DEFAULT_VENV_DIR

Valeur: join(...)

nemo_rl.utils.venvs.logger

Valeur: getLogger(...)

nemo_rl.utils.venvs.create_local_venv(
    py_executable: str,
    venv_name: str,
    force_rebuild: bool = False
) -> str

Créer un environnement virtuel en utilisant uv et exécuter une commande dans celui-ci.

La sortie peut être utilisée comme py_executable pour un worker Ray en supposant que les nœuds workers ont également accès au même système de fichiers que le nœud principal.

Cette fonction est mise en cache pour éviter plusieurs appels à uv pour créer le même venv, ce qui évite les journaux en double.

Arguments : py_executable (str) : Commande à exécuter avec l’environnement virtuel (par exemple, “uv.sh run —locked”) venv_name (str) : Nom de l’environnement virtuel (par exemple, “foobar.Worker”) force_rebuild (bool) : Si True, forcer la reconstruction du venv même s’il existe déjà

Retourne : str : Chemin de l’exécutable Python dans l’environnement virtuel créé

nemo_rl.utils.venvs._env_builder(
    py_executable: str,
    venv_name: str,
    node_idx: int,
    force_rebuild: bool = False
)

nemo_rl.utils.venvs.create_local_venv_on_each_node(
    py_executable: str, venv_name: str
)

Créer un environnement virtuel sur chaque nœud Ray.

Arguments : py_executable (str) : Commande à exécuter avec l’environnement virtuel venv_name (str) : Nom de l’environnement virtuel

Retourne : str : Chemin de l’exécutable Python dans l’environnement virtuel créé