INNER CODE UNIT · Python
TrajectoryGenerator
Future-House/BixBench · bixbench/generate_trajectories.py:42
class TrajectoryGenerator:
"""
Generator for creating and storing agent trajectories on data analysis tasks.
This class handles the full pipeline of loading benchmark capsules, setting up
environments, running agents through these environments, and storing the resulting
trajectories.
"""
def __init__(
self, config_path=DEFAULT_CONFIG_PATH, replica_id: int | None = None
) -> None:
"""
Initialize the TrajectoryGenerator with config and create necessary directories.
Args:
config_path: Path to the configuration file
replica_id: Replica ID