Function load_trajectories

src/evaluation/loader.py:15–33

Load every ``*.json`` trajectory under ``path``. ``path`` may be a directory or a single JSON file. If a trajectory has ``scenario_id`` set to null, the filename stem is used as a fallback. This supports layouts such as ``traces/trajectories/34.json`` where ``34`` is the scenario id

(path: Path)

def load_trajectories(path: Path) -> list[PersistedTrajectory]:
    """Load every ``*.json`` trajectory under ``path``.

    ``path`` may be a directory or a single JSON file. If a trajectory has
    ``scenario_id`` set to null, the filename stem is used as a fallback.
    This supports layouts such as ``traces/trajectories/34.json`` where
    ``34`` is the scenario id.
    """
    p = Path(path)
    if p.is_file():
        return [_load_one_trajectory(p)] if p.suffix == ".json" else []

    out: list[PersistedTrajectory] = []
    for child in sorted(p.glob("*.json")):
        try:
            out.append(_load_one_trajectory(child))
        except Exception:
            _log.exception("loader: failed to parse %s", child)
    return out


def _load_one_trajectory(path: Path) -> PersistedTrajectory: