title: CelebA Dataset inaccessible using tfds.load()
body: <p>I am trying to use the CelebA dataset in a deep learning project. I have the zipped folder from Kaggle.
  I wanted to unzip and then split the images into training, testing, and validation, but then found out that it would not be possible on my <em>not-so-powerful</em> system.</p>
  <p>So, to avoid wasting time, I wanted to use the TensorFlow-datasets method to load the CelebA dataset. But unfortunately, the dataset is inaccessible with the following error:</p>
  <p><strong>(Code first)</strong></p>
  <pre><code>ds = tfds.load('celeb_a', split='train', download=True)
  </code></pre>
  <pre><code>---------------------------------------------------------------------------
  KeyError                                  Traceback (most recent call last)
  &lt;ipython-input-69-d7b9371eb674&gt; in &lt;module&gt;
  ----&gt; 1 ds = tfds.load('celeb_a', split='train', download=True)
  
  c:\users\aman\appdata\local\programs\python\python38\lib\site-packages\tensorflow_datasets\core\load.py in load(name, split, data_dir, batch_size, shuffle_files, download, as_supervised, decoders, read_config, with_info, builder_kwargs, download_and_prepare_kwargs, as_dataset_kwargs, try_gcs)
      344   if download:
      345     download_and_prepare_kwargs = download_and_prepare_kwargs or {}
  --&gt; 346     dbuilder.download_and_prepare(**download_and_prepare_kwargs)
      347 
      348   if as_dataset_kwargs is None:
  
  c:\users\aman\appdata\local\programs\python\python38\lib\site-packages\tensorflow_datasets\core\dataset_builder.py in download_and_prepare(self, download_dir, download_config)
      383           self.info.read_from_directory(self._data_dir)
      384         else:
  --&gt; 385           self._download_and_prepare(
      386               dl_manager=dl_manager,
      387               download_config=download_config)
  
  c:\users\aman\appdata\local\programs\python\python38\lib\site-packages\tensorflow_datasets\core\dataset_builder.py in _download_and_prepare(self, dl_manager, download_config)
     1020   def _download_and_prepare(self, dl_manager, download_config):
     1021     # Extract max_examples_per_split and forward it to _prepare_split
  -&gt; 1022     super(GeneratorBasedBuilder, self)._download_and_prepare(
     1023         dl_manager=dl_manager,
     1024         max_examples_per_split=download_config.max_examples_per_split,
  
  c:\users\aman\appdata\local\programs\python\python38\lib\site-packages\tensorflow_datasets\core\dataset_builder.py in _download_and_prepare(self, dl_manager, **prepare_split_kwargs)
      959     split_generators_kwargs = self._make_split_generators_kwargs(
      960         prepare_split_kwargs)
  --&gt; 961     for split_generator in self._split_generators(
      962         dl_manager, **split_generators_kwargs):
      963       if str(split_generator.split_info.name).lower() == &quot;all&quot;:
  
  c:\users\aman\appdata\local\programs\python\python38\lib\site-packages\tensorflow_datasets\image\celeba.py in _split_generators(self, dl_manager)
      137     all_images = {
      138         os.path.split(k)[-1]: img for k, img in
  --&gt; 139         dl_manager.iter_archive(downloaded_dirs[&quot;img_align_celeba&quot;])
      140     }
      141 
  
  c:\users\aman\appdata\local\programs\python\python38\lib\site-packages\tensorflow_datasets\core\download\download_manager.py in iter_archive(self, resource)
      559     if isinstance(resource, six.string_types):
      560       resource = resource_lib.Resource(path=resource)
  --&gt; 561     return extractor.iter_archive(resource.path, resource.extract_method)
      562 
      563   def extract(self, path_or_paths):
  
  c:\users\aman\appdata\local\programs\python\python38\lib\site-packages\tensorflow_datasets\core\download\extractor.py in iter_archive(path, method)
      221     An iterator of `(path_in_archive, f_obj)`
      222   &quot;&quot;&quot;
  --&gt; 223   return _EXTRACT_METHODS[method](path)
  
  KeyError: &lt;ExtractMethod.NO_EXTRACT: 1&gt;
  </code></pre>
  <p>Could someone explain what I am doing wrong?</p>
  <p>On a side-note, if this does not work, is there a way to convert the already downloaded zipped file from Kaggle into the required format without unzipping and then iterating over each image individually? Basically, I cannot go down the <em>unzip-then-split</em> route for such a large dataset...</p>
  <p>TIA!</p>
  <hr />
  <p><em>EDIT:</em> I tried the same on Colab, but getting a similar error:
  <a href="https://i.stack.imgur.com/uNySQ.png" rel="noreferrer"><img src="https://i.stack.imgur.com/uNySQ.png" alt="enter image description here" /></a></p>