"""Dataset generation and loading utilities."""