INNER CODE UNIT · Python

filter_completed

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

    def filter_completed(self, bixbench: list[dict[str, Any]]) -> list[dict[str, Any]]:
        """Filter out completed trajectories."""
        return [
            question
            for question in bixbench
            if not self.get_trajectory_path(question["question_id"]).exists()
        ]

    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:

View source record →