from pathlib import Path
from spectral_world_models.data import DatasetConfig, generate_benchmark_npz

if __name__ == "__main__":
    out = Path(__file__).resolve().parent / "mini_moving_shapes.npz"
    generate_benchmark_npz(out, DatasetConfig())
    print(f"wrote {out}")