"""Offline data utilities for the Self-Forcing Predictor v4."""

from .capture import PredictorV4TeacherCapture
from .schema import (
    CANDIDATE_BLOCK_IDS,
    CHUNK_FRAMES,
    NUM_CHUNKS,
    NUM_STEPS,
    SCHEMA_VERSION,
    validate_case_tensors,
    validate_clean_prefeature,
    validate_step_tensors,
)
from .writer import PredictorV4DatasetWriter

__all__ = [
    "CANDIDATE_BLOCK_IDS",
    "CHUNK_FRAMES",
    "NUM_CHUNKS",
    "NUM_STEPS",
    "SCHEMA_VERSION",
    "PredictorV4DatasetWriter",
    "PredictorV4TeacherCapture",
    "validate_case_tensors",
    "validate_clean_prefeature",
    "validate_step_tensors",
]