import tensorflow_datasets as tfds
tfds.load('celeb_a')

KeyError                                  Traceback (most recent call last)
<ipython-input-22-c041a8b6c170> in <module>
      4 # datasets = builder.as_dataset()
      5 
----> 6 tfds.load('celeb_a', split=tfds.Split.TRAIN)
      7 # tfds.load('celeb_a', split="train")

~/labo/generative_deep_learning/.venv/lib/python3.8/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)
    342   if download:
    343     download_and_prepare_kwargs = download_and_prepare_kwargs or {}
--> 344     dbuilder.download_and_prepare(**download_and_prepare_kwargs)
    345 
    346   if as_dataset_kwargs is None:

~/labo/generative_deep_learning/.venv/lib/python3.8/site-packages/tensorflow_datasets/core/dataset_builder.py in download_and_prepare(self, download_dir, download_config)
    416           # Old version of TF are not os.PathLike compatible
    417           with tf_compat.mock_gfile_pathlike():
--> 418             self._download_and_prepare(
    419                 dl_manager=dl_manager,
    420                 download_config=download_config,

~/labo/generative_deep_learning/.venv/lib/python3.8/site-packages/tensorflow_datasets/core/dataset_builder.py in _download_and_prepare(self, dl_manager, download_config)
   1093       else:
   1094         optional_pipeline_kwargs = {}
-> 1095       split_generators = self._split_generators(  # pylint: disable=unexpected-keyword-arg
   1096           dl_manager, **optional_pipeline_kwargs
   1097       )

~/labo/generative_deep_learning/.venv/lib/python3.8/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
--> 139         dl_manager.iter_archive(downloaded_dirs["img_align_celeba"])
    140     }
    141 

~/labo/generative_deep_learning/.venv/lib/python3.8/site-packages/tensorflow_datasets/core/download/download_manager.py in iter_archive(self, resource)
    663     if isinstance(resource, type_utils.PathLikeCls):
    664       resource = resource_lib.Resource(path=resource)
--> 665     return extractor.iter_archive(resource.path, resource.extract_method)
    666 
    667   @typing.overload

~/labo/generative_deep_learning/.venv/lib/python3.8/site-packages/tensorflow_datasets/core/download/extractor.py in iter_archive(path, method)
    224     An iterator of `(path_in_archive, f_obj)`
    225   """
--> 226   return _EXTRACT_METHODS[method](path)

KeyError: <ExtractMethod.NO_EXTRACT: 1>