Как загрузить часть tensorflow dataset lsun?

Не могу загрузить только часть датасета lsun, потому что в версии 3.0.0 работал split api, а в 3.1.0 нужно добавить конфигурацию builder для категории объектов missing person и добавьте идентификатор в словарь объектов (3.1.0 (default): Add builder config for missing person object category, and add id to the feature dict) и я не пойму как это сделать.

https://www.tensorflow.org/datasets/catalog/lsun