nemo_gym.rollout_collection#

Module Contents#

Classes#

RolloutCollectionConfig

Perform a batch of rollout collection.

RolloutCollectionHelper

Functions#

collect_rollouts

API#

class nemo_gym.rollout_collection.RolloutCollectionConfig#

Bases: nemo_gym.config_types.BaseNeMoGymCLIConfig

Perform a batch of rollout collection.

agent_name: str#

‘Field(…)’

input_jsonl_fpath: str#

‘Field(…)’

output_jsonl_fpath: str#

‘Field(…)’

limit: Optional[int]#

‘Field(…)’

num_repeats: Optional[int]#

‘Field(…)’

num_samples_in_parallel: Optional[int]#

‘Field(…)’

responses_create_params: Dict[str, Any]#

‘Field(…)’

class nemo_gym.rollout_collection.RolloutCollectionHelper#

Bases: pydantic.BaseModel

async run_from_config(
config: nemo_gym.rollout_collection.RolloutCollectionConfig,
)#
run_examples(
examples: List[Dict],
head_server_config: Optional[nemo_gym.config_types.BaseServerConfig] = None,
) → Iterator[asyncio.Future]#

We provide this function as a lower level interface for running rollout collection.

setup_server_client(
head_server_config: Optional[nemo_gym.config_types.BaseServerConfig] = None,
) → nemo_gym.server_utils.ServerClient#
nemo_gym.rollout_collection.collect_rollouts()#