code: def _split_generators(self, dl_manager):
      
      url = _DL_URLS[self.builder_config.name]
      data_dirs = dl_manager.download_and_extract(url)
  
      path_to_dataset = os.path.join(data_dirs, tf.io.gfile.listdir(data_dirs)[0])
  
      train_a_path = os.path.join(path_to_dataset, "trainA")
      train_b_path = os.path.join(path_to_dataset, "trainB")
      test_a_path = os.path.join(path_to_dataset, "testA")
      test_b_path = os.path.join(path_to_dataset, "testB")
  
      return [
          tfds.core.SplitGenerator(
              name="trainA",
              num_shards=10,
              gen_kwargs={
                  "path": train_a_path,
                  "label": "A",
              }),
          tfds.core.SplitGenerator(
              name="trainB",
              num_shards=10,
              gen_kwargs={
                  "path": train_b_path,
                  "label": "B",
              }),
          tfds.core.SplitGenerator(
              name="testA",
              num_shards=1,
              gen_kwargs={
                  "path": test_a_path,
                  "label": "A",
              }),
          tfds.core.SplitGenerator(
              name="testB",
              num_shards=1,
              gen_kwargs={
                  "path": test_b_path,
                  "label": "B",
              }),
      ]
docs: Returns SplitGenerators.