INNER CODE UNIT · Python

rollout_with_saving

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

    async def rollout_with_saving(
        self, environment: DataAnalysisEnv
    ) -> list[tuple[Trajectory, DataAnalysisEnv]]:
        trajectory, env = await self.vanilla_rollout(
            self.config.agent_config.construct_agent(), environment
        )
        await self.store_trajectory(trajectory, env)
        return trajectory, env

    async def run(self) -> None:
        """Run the full trajectory generation pipeline."""
        logger.info("Loading BixBench dataset...")
        bixbench = await self.load_bixbench()

        if self.config.rollout.skip_existing_trajectories:
            bixbench = self.filter_completed(bixbench)

        # Process environments in batches with tqdm progress bar

View source record →