INNER CODE UNIT · Python

vanilla_rollout

Future-House/BixBench · bixbench/generate_trajectories.py:352

    async def vanilla_rollout(
        self, agent: Agent, environment: DataAnalysisEnv
    ) -> tuple[Trajectory, DataAnalysisEnv]:
        """
        Standard implementation of rollout logic.

        Args:
            agent: The agent to use for rollout
            environment: The environment to run the agent in

        Returns:
            Tuple[Trajectory, DataAnalysisEnv]: The generated trajectory and updated environment
        """
        obs, tools = await environment.reset()
        agent_state = await agent.init_state(tools)
        trajectory = Trajectory()

        for timestep in range(self.config.rollout.max_steps):

View source record →