nemo_rl.environments.utils

Contenu du Module

Fonctions

API

nemo_rl.environments.utils.chunk_list_to_workers(
    to_chunk: list[typing.Any], num_workers: int
) -> list[list[typing.Any]]

Diviser une liste en une liste de listes, où chaque sous-liste est assignée à un worker. Préserve l’ordre des éléments.

Si la liste n’est pas divisible par le nombre de workers, le dernier worker peut avoir moins d’éléments. S’il y a plus de workers que d’éléments, les premiers len(list) workers auront un élément chacun, et les workers restants auront des listes vides.

Arguments : list : La liste à diviser. num_workers : Le nombre de workers auquel distribuer la liste.

Retourne : Une liste de listes, où chaque sous-liste contient des éléments assignés à un worker.

Exemples :

>>> from nemo_rl.environments.utils import chunk_list_to_workers
>>> chunk_list_to_workers([1, 2, 3, 4, 5], 3)
[[1, 2], [3, 4], [5]]