INNER CODE UNIT · Python

custom_rollout

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

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

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

        Returns:
            Trajectory: The generated trajectory

        Raises:
            NotImplementedError: This method needs to be implemented by subclasses
        """
        raise NotImplementedError("Custom rollout not implemented")

View source record →