File size: 274 Bytes
4fd79a1 | 1 2 3 4 5 6 7 8 | 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}")
|