class_docstr: CelebA dataset. Aligned and cropped. With metadata.
class_docstr_tok: ["CelebA","dataset",".","Aligned","and","cropped",".","With","metadata","."]
code: def _split_generators(self, dl_manager):
      downloaded_dirs = dl_manager.download(
          {
              "img_align_celeba": IMG_ALIGNED_DATA,
              "list_eval_partition": EVAL_LIST,
              "list_attr_celeba": ATTR_DATA,
              "landmarks_celeba": LANDMARKS_DATA,
              "identity_celeba": IDENTITY_DATA,
          }
      )
  
      # Load all images in memory (~1 GiB)
      # Use split to convert: `img_align_celeba/000005.jpg` -> `000005.jpg`
      all_images = {
          os.path.split(k)[-1]: img
          for k, img in dl_manager.iter_archive(
              downloaded_dirs["img_align_celeba"]
          )
      }
  
      return [
          tfds.core.SplitGenerator(
              name=tfds.Split.TRAIN,
              gen_kwargs={
                  "file_id": 0,
                  "downloaded_dirs": downloaded_dirs,
                  "downloaded_images": all_images,
              },
          ),
          tfds.core.SplitGenerator(
              name=tfds.Split.VALIDATION,
              gen_kwargs={
                  "file_id": 1,
                  "downloaded_dirs": downloaded_dirs,
                  "downloaded_images": all_images,
              },
          ),
          tfds.core.SplitGenerator(
              name=tfds.Split.TEST,
              gen_kwargs={
                  "file_id": 2,
                  "downloaded_dirs": downloaded_dirs,
                  "downloaded_images": all_images,
              },
          ),
      ]
  
file_path: datasets/tensorflow_datasets/datasets/celeb_a/celeb_a_dataset_builder.py
imports: from __future__ import annotations
  import os
  from typing import Sequence, Union
  from etils import epath
  import numpy
  import tensorflow_datasets.public_api
label_desc: Use image node_id 2 for calling the Builder obj's underlying member method code with example usage: obj._split_generators(dl_manager) and returns: list