INNER CODE UNIT · Python

store_trajectory

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

    async def store_trajectory(
        self, trajectory: Trajectory, env: DataAnalysisEnv
    ) -> None:
        """
        Store trajectory and environment information to disk.

        Args:
            trajectory: The trajectory to store
            env: The environment that generated the trajectory
        """
        metadata = (
            {k: v for k, v in env.metadata.items() if k != "local_data_folder"}
            if env.metadata
            else {}
        )
        mcqs = metadata.pop("mcqs", [])
        mcq_options = mcqs[0].options if mcqs else []
        mcq_question = mcqs[0].question if mcqs else ""

View source record →