INNER CODE UNIT · Python
batch_rollout
Future-House/BixBench · bixbench/generate_trajectories.py:395
async def batch_rollout(
self, list_of_environments: list[DataAnalysisEnv]
) -> list[tuple[Trajectory, DataAnalysisEnv]]:
"""
Run rollouts for a batch of environments.
Args:
list_of_environments: List of environments to run rollouts in
Returns:
List[Union[Trajectory, Tuple[Trajectory, DataAnalysisEnv]]]: List of trajectories or
trajectory-environment pairs depending on rollout type
"""
if self.config.rollout.rollout_type == "aviary":
agent = self.config.agent_config.construct_agent()
rollout = RolloutManager(agent=agent)
trajectories = await rollout.sample_trajectories(
environments=list_of_environments,