tfds-nightly 4.9.9.dev202506020045__tar.gz → 4.9.9.dev202506030044__tar.gz

tfds_nightly-4.9.9.dev202506030044/PKG-INFO

@@ -0,0 +1,276 @@
+Metadata-Version: 2.4
+Name: tfds-nightly
+Version: 4.9.9.dev202506030044
+Summary: tensorflow/datasets is a library of datasets ready to use with TensorFlow.
+Home-page: https://github.com/tensorflow/datasets
+Download-URL: https://github.com/tensorflow/datasets/tags
+Author: Google Inc.
+Author-email: packages@tensorflow.org
+License: Apache 2.0
+Keywords: tensorflow machine learning datasets
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10
+License-File: LICENSE
+License-File: AUTHORS
+Requires-Dist: absl-py
+Requires-Dist: array_record>=0.5.0; platform_system == "Linux"
+Requires-Dist: dm-tree
+Requires-Dist: etils[edc,enp,epath,epy,etree]>=1.6.0; python_version < "3.11"
+Requires-Dist: etils[edc,enp,epath,epy,etree]>=1.9.1; python_version >= "3.11"
+Requires-Dist: immutabledict
+Requires-Dist: numpy
+Requires-Dist: promise
+Requires-Dist: protobuf>=3.20
+Requires-Dist: psutil
+Requires-Dist: pyarrow
+Requires-Dist: requests>=2.19.0
+Requires-Dist: simple_parsing
+Requires-Dist: tensorflow-metadata
+Requires-Dist: termcolor
+Requires-Dist: toml
+Requires-Dist: tqdm
+Requires-Dist: wrapt
+Requires-Dist: importlib_resources; python_version < "3.9"
+Provides-Extra: matplotlib
+Requires-Dist: matplotlib; extra == "matplotlib"
+Provides-Extra: tensorflow
+Requires-Dist: tensorflow>=2.1; extra == "tensorflow"
+Provides-Extra: tf-nightly
+Requires-Dist: tf-nightly; extra == "tf-nightly"
+Provides-Extra: tensorflow-data-validation
+Requires-Dist: tensorflow-data-validation; extra == "tensorflow-data-validation"
+Provides-Extra: tests-all
+Requires-Dist: dill; extra == "tests-all"
+Requires-Dist: jax[cpu]==0.4.28; extra == "tests-all"
+Requires-Dist: jupyter; extra == "tests-all"
+Requires-Dist: pytest; extra == "tests-all"
+Requires-Dist: pytest-shard; extra == "tests-all"
+Requires-Dist: pytest-xdist; extra == "tests-all"
+Requires-Dist: apache-beam<2.65.0; extra == "tests-all"
+Requires-Dist: conllu; extra == "tests-all"
+Requires-Dist: mlcroissant>=1.0.9; extra == "tests-all"
+Requires-Dist: pandas; extra == "tests-all"
+Requires-Dist: pydub; extra == "tests-all"
+Requires-Dist: pyyaml; extra == "tests-all"
+Requires-Dist: tensorflow-io[tensorflow]; python_version < "3.12" and extra == "tests-all"
+Requires-Dist: scipy; extra == "tests-all"
+Requires-Dist: apache-beam<2.65.0; extra == "tests-all"
+Requires-Dist: gcsfs; extra == "tests-all"
+Requires-Dist: zarr<3.0.0; extra == "tests-all"
+Requires-Dist: apache-beam<2.65.0; extra == "tests-all"
+Requires-Dist: gcld3; extra == "tests-all"
+Requires-Dist: langdetect; extra == "tests-all"
+Requires-Dist: nltk==3.8.1; extra == "tests-all"
+Requires-Dist: tldextract; extra == "tests-all"
+Requires-Dist: apache-beam<2.65.0; extra == "tests-all"
+Requires-Dist: matplotlib; extra == "tests-all"
+Requires-Dist: Pillow; extra == "tests-all"
+Requires-Dist: pydub; extra == "tests-all"
+Requires-Dist: scipy; extra == "tests-all"
+Requires-Dist: scikit-image; extra == "tests-all"
+Requires-Dist: tifffile; extra == "tests-all"
+Requires-Dist: imagecodecs; extra == "tests-all"
+Requires-Dist: pretty_midi; extra == "tests-all"
+Requires-Dist: pydub; extra == "tests-all"
+Requires-Dist: pydub; extra == "tests-all"
+Requires-Dist: opencv-python; extra == "tests-all"
+Requires-Dist: scikit-image; extra == "tests-all"
+Requires-Dist: scipy; extra == "tests-all"
+Requires-Dist: pydub; extra == "tests-all"
+Requires-Dist: pandas; extra == "tests-all"
+Requires-Dist: networkx; extra == "tests-all"
+Requires-Dist: pandas; extra == "tests-all"
+Requires-Dist: pandas; extra == "tests-all"
+Requires-Dist: h5py; extra == "tests-all"
+Requires-Dist: envlogger; (python_version < "3.11" and sys_platform == "linux") and extra == "tests-all"
+Requires-Dist: envlogger; (python_version < "3.11" and sys_platform == "linux") and extra == "tests-all"
+Requires-Dist: pandas; extra == "tests-all"
+Requires-Dist: pydub; extra == "tests-all"
+Requires-Dist: scipy; extra == "tests-all"
+Requires-Dist: scipy; extra == "tests-all"
+Requires-Dist: pandas; extra == "tests-all"
+Requires-Dist: Pillow; extra == "tests-all"
+Requires-Dist: apache-beam<2.65.0; extra == "tests-all"
+Requires-Dist: apache-beam<2.65.0; extra == "tests-all"
+Requires-Dist: mwparserfromhell; extra == "tests-all"
+Requires-Dist: mwxml; extra == "tests-all"
+Requires-Dist: bs4; extra == "tests-all"
+Requires-Dist: lxml; extra == "tests-all"
+Requires-Dist: pycocotools; extra == "tests-all"
+Provides-Extra: dev
+Requires-Dist: dill; extra == "dev"
+Requires-Dist: jax[cpu]==0.4.28; extra == "dev"
+Requires-Dist: jupyter; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-shard; extra == "dev"
+Requires-Dist: pytest-xdist; extra == "dev"
+Requires-Dist: apache-beam<2.65.0; extra == "dev"
+Requires-Dist: conllu; extra == "dev"
+Requires-Dist: mlcroissant>=1.0.9; extra == "dev"
+Requires-Dist: pandas; extra == "dev"
+Requires-Dist: pydub; extra == "dev"
+Requires-Dist: pyyaml; extra == "dev"
+Requires-Dist: tensorflow-io[tensorflow]; python_version < "3.12" and extra == "dev"
+Requires-Dist: pylint>=2.6.0; extra == "dev"
+Requires-Dist: yapf; extra == "dev"
+Provides-Extra: huggingface
+Requires-Dist: dill; extra == "huggingface"
+Requires-Dist: jax[cpu]==0.4.28; extra == "huggingface"
+Requires-Dist: jupyter; extra == "huggingface"
+Requires-Dist: pytest; extra == "huggingface"
+Requires-Dist: pytest-shard; extra == "huggingface"
+Requires-Dist: pytest-xdist; extra == "huggingface"
+Requires-Dist: conllu; extra == "huggingface"
+Requires-Dist: mlcroissant>=1.0.9; extra == "huggingface"
+Requires-Dist: pandas; extra == "huggingface"
+Requires-Dist: pydub; extra == "huggingface"
+Requires-Dist: pyyaml; extra == "huggingface"
+Requires-Dist: tensorflow-io[tensorflow]; python_version < "3.12" and extra == "huggingface"
+Requires-Dist: scipy; extra == "huggingface"
+Requires-Dist: gcsfs; extra == "huggingface"
+Requires-Dist: zarr<3.0.0; extra == "huggingface"
+Requires-Dist: gcld3; extra == "huggingface"
+Requires-Dist: langdetect; extra == "huggingface"
+Requires-Dist: nltk==3.8.1; extra == "huggingface"
+Requires-Dist: tldextract; extra == "huggingface"
+Requires-Dist: matplotlib; extra == "huggingface"
+Requires-Dist: Pillow; extra == "huggingface"
+Requires-Dist: pydub; extra == "huggingface"
+Requires-Dist: scipy; extra == "huggingface"
+Requires-Dist: scikit-image; extra == "huggingface"
+Requires-Dist: tifffile; extra == "huggingface"
+Requires-Dist: imagecodecs; extra == "huggingface"
+Requires-Dist: pretty_midi; extra == "huggingface"
+Requires-Dist: pydub; extra == "huggingface"
+Requires-Dist: pydub; extra == "huggingface"
+Requires-Dist: opencv-python; extra == "huggingface"
+Requires-Dist: scikit-image; extra == "huggingface"
+Requires-Dist: scipy; extra == "huggingface"
+Requires-Dist: pydub; extra == "huggingface"
+Requires-Dist: pandas; extra == "huggingface"
+Requires-Dist: networkx; extra == "huggingface"
+Requires-Dist: pandas; extra == "huggingface"
+Requires-Dist: pandas; extra == "huggingface"
+Requires-Dist: h5py; extra == "huggingface"
+Requires-Dist: envlogger; (python_version < "3.11" and sys_platform == "linux") and extra == "huggingface"
+Requires-Dist: envlogger; (python_version < "3.11" and sys_platform == "linux") and extra == "huggingface"
+Requires-Dist: pandas; extra == "huggingface"
+Requires-Dist: pydub; extra == "huggingface"
+Requires-Dist: scipy; extra == "huggingface"
+Requires-Dist: scipy; extra == "huggingface"
+Requires-Dist: pandas; extra == "huggingface"
+Requires-Dist: Pillow; extra == "huggingface"
+Requires-Dist: mwparserfromhell; extra == "huggingface"
+Requires-Dist: mwxml; extra == "huggingface"
+Requires-Dist: bs4; extra == "huggingface"
+Requires-Dist: lxml; extra == "huggingface"
+Requires-Dist: pycocotools; extra == "huggingface"
+Requires-Dist: datasets; extra == "huggingface"
+Provides-Extra: aflw2k3d
+Requires-Dist: scipy; extra == "aflw2k3d"
+Provides-Extra: beir
+Requires-Dist: apache-beam<2.65.0; extra == "beir"
+Provides-Extra: ble-wind-field
+Requires-Dist: gcsfs; extra == "ble-wind-field"
+Requires-Dist: zarr<3.0.0; extra == "ble-wind-field"
+Provides-Extra: c4
+Requires-Dist: apache-beam<2.65.0; extra == "c4"
+Requires-Dist: gcld3; extra == "c4"
+Requires-Dist: langdetect; extra == "c4"
+Requires-Dist: nltk==3.8.1; extra == "c4"
+Requires-Dist: tldextract; extra == "c4"
+Provides-Extra: c4-wsrs
+Requires-Dist: apache-beam<2.65.0; extra == "c4-wsrs"
+Provides-Extra: cats-vs-dogs
+Requires-Dist: matplotlib; extra == "cats-vs-dogs"
+Provides-Extra: colorectal-histology
+Requires-Dist: Pillow; extra == "colorectal-histology"
+Provides-Extra: common-voice
+Requires-Dist: pydub; extra == "common-voice"
+Provides-Extra: duke-ultrasound
+Requires-Dist: scipy; extra == "duke-ultrasound"
+Provides-Extra: eurosat
+Requires-Dist: scikit-image; extra == "eurosat"
+Requires-Dist: tifffile; extra == "eurosat"
+Requires-Dist: imagecodecs; extra == "eurosat"
+Provides-Extra: groove
+Requires-Dist: pretty_midi; extra == "groove"
+Requires-Dist: pydub; extra == "groove"
+Provides-Extra: gtzan
+Requires-Dist: pydub; extra == "gtzan"
+Provides-Extra: imagenet2012-corrupted
+Requires-Dist: opencv-python; extra == "imagenet2012-corrupted"
+Requires-Dist: scikit-image; extra == "imagenet2012-corrupted"
+Requires-Dist: scipy; extra == "imagenet2012-corrupted"
+Provides-Extra: librispeech
+Requires-Dist: pydub; extra == "librispeech"
+Provides-Extra: lsun
+Requires-Dist: tensorflow-io[tensorflow]; extra == "lsun"
+Provides-Extra: nsynth
+Requires-Dist: crepe>=0.0.11; extra == "nsynth"
+Requires-Dist: librosa; extra == "nsynth"
+Requires-Dist: scikit-learn==0.20.3; extra == "nsynth"
+Provides-Extra: ogbg-molpcba
+Requires-Dist: pandas; extra == "ogbg-molpcba"
+Requires-Dist: networkx; extra == "ogbg-molpcba"
+Provides-Extra: pet-finder
+Requires-Dist: pandas; extra == "pet-finder"
+Provides-Extra: qm9
+Requires-Dist: pandas; extra == "qm9"
+Provides-Extra: robonet
+Requires-Dist: h5py; extra == "robonet"
+Provides-Extra: locomotion
+Requires-Dist: envlogger; (python_version < "3.11" and sys_platform == "linux") and extra == "locomotion"
+Provides-Extra: robosuite-panda-pick-place-can
+Requires-Dist: envlogger; (python_version < "3.11" and sys_platform == "linux") and extra == "robosuite-panda-pick-place-can"
+Provides-Extra: smartwatch-gestures
+Requires-Dist: pandas; extra == "smartwatch-gestures"
+Provides-Extra: speech-commands
+Requires-Dist: pydub; extra == "speech-commands"
+Provides-Extra: svhn
+Requires-Dist: scipy; extra == "svhn"
+Provides-Extra: the300w-lp
+Requires-Dist: scipy; extra == "the300w-lp"
+Provides-Extra: wake-vision
+Requires-Dist: pandas; extra == "wake-vision"
+Provides-Extra: wider-face
+Requires-Dist: Pillow; extra == "wider-face"
+Provides-Extra: wiki-dialog
+Requires-Dist: apache-beam<2.65.0; extra == "wiki-dialog"
+Provides-Extra: wikipedia
+Requires-Dist: apache-beam<2.65.0; extra == "wikipedia"
+Requires-Dist: mwparserfromhell; extra == "wikipedia"
+Requires-Dist: mwxml; extra == "wikipedia"
+Provides-Extra: wsc273
+Requires-Dist: bs4; extra == "wsc273"
+Requires-Dist: lxml; extra == "wsc273"
+Provides-Extra: youtube-vis
+Requires-Dist: pycocotools; extra == "youtube-vis"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: download-url
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+tensorflow/datasets is a library of public datasets ready to use with
+TensorFlow. Each dataset definition contains the logic necessary to download and
+prepare the dataset, as well as to read it into a model using the
+`tf.data.Dataset` API.
+
+Usage outside of TensorFlow is also supported.
+
+See the README on GitHub for further documentation.

tfds_nightly-4.9.9.dev202506030044/setup.py

@@ -0,0 +1,282 @@
+# coding=utf-8
+# Copyright 2025 The TensorFlow Datasets Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""tensorflow/datasets is a library of datasets ready to use with TensorFlow.
+
+tensorflow/datasets is a library of public datasets ready to use with
+TensorFlow. Each dataset definition contains the logic necessary to download and
+prepare the dataset, as well as to read it into a model using the
+`tf.data.Dataset` API.
+
+Usage outside of TensorFlow is also supported.
+
+See the README on GitHub for further documentation.
+"""
+
+import itertools
+import os
+import sys
+
+import pkg_resources
+import setuptools
+
+# To enable importing version.py directly, we add its path to sys.path.
+version_path = os.path.join(os.path.dirname(__file__), 'tensorflow_datasets')
+sys.path.append(version_path)
+from version import __version__  # pytype: disable=import-error  # pylint: disable=g-import-not-at-top
+
+if datestring := os.environ.get('TFDS_NIGHTLY_TIMESTAMP'):
+  project_name = 'tfds-nightly'
+  # Version as `X.Y.Z.dev199912312459`
+  curr_version = pkg_resources.parse_version(__version__)
+  __version__ = f'{curr_version.base_version}.dev{datestring}'
+else:
+  project_name = 'tensorflow-datasets'
+
+DOCLINES = __doc__.split('\n')
+
+REQUIRED_PKGS = [
+    'absl-py',
+    # Min version of 0.5.0 as old array_record wheel are bugged on all
+    # platform except 'x86_64'. See
+    # https://github.com/google/array_record/issues/71
+    'array_record>=0.5.0;platform_system=="Linux"',
+    'dm-tree',
+    'etils[edc,enp,epath,epy,etree]>=1.6.0;python_version<"3.11"',
+    'etils[edc,enp,epath,epy,etree]>=1.9.1;python_version>="3.11"',
+    'immutabledict',
+    'numpy',
+    'promise',
+    'protobuf>=3.20',
+    'psutil',
+    'pyarrow',
+    'requests>=2.19.0',
+    'simple_parsing',
+    'tensorflow-metadata',
+    'termcolor',
+    'toml',
+    'tqdm',
+    'wrapt',
+    # Standard library backports
+    'importlib_resources;python_version<"3.9"',
+]
+
+TESTS_DEPENDENCIES = [
+    'dill',
+    'jax[cpu]==0.4.28',
+    'jupyter',
+    'pytest',
+    'pytest-shard',
+    'pytest-xdist',
+    # Lazy-deps required by core
+    # TODO(b/418761065): Update to 2.65.0 once the bug is fixed.
+    'apache-beam<2.65.0',
+    'conllu',
+    'mlcroissant>=1.0.9',
+    'pandas',
+    'pydub',
+    # Required by scripts/documentation/
+    'pyyaml',
+    'tensorflow-io[tensorflow];python_version<"3.12"',
+]
+
+# Additional deps for formatting
+DEV_DEPENDENCIES = [
+    'pylint>=2.6.0',
+    'yapf',
+]
+
+# Static files needed by datasets.
+DATASET_FILES = [
+    'datasets/imagenet2012/labels.txt',
+    'datasets/imagenet2012/validation_labels.txt',
+    'datasets/lvis/classes.txt',
+    'datasets/ogbg_molpcba/tasks.txt',
+    'datasets/quickdraw/labels.txt',
+    'datasets/smartwatch_gestures/labels.txt',
+    'image_classification/caltech101_labels.txt',
+    'image_classification/categories_places365.txt',
+    'image_classification/cbis_ddsm_calc_distributions.txt',
+    'image_classification/cbis_ddsm_calc_types.txt',
+    'image_classification/cbis_ddsm_mass_margins.txt',
+    'image_classification/cbis_ddsm_mass_shapes.txt',
+    'image_classification/cbis_ddsm_patch_labels.txt',
+    'image_classification/dtd_key_attributes.txt',
+    'image_classification/food-101_classes.txt',
+    'image_classification/i_naturalist2018/inaturalist2018_labels.txt',
+    'image_classification/i_naturalist2018/inaturalist2018_supercategories.txt',
+    'image_classification/i_naturalist2021/i_naturalist2021_labels.txt',
+    'image_classification/i_naturalist2021/i_naturalist2021_supercategories.txt',
+    'image_classification/imagenet_resized_labels.txt',
+    'image_classification/imagenette_labels.txt',
+    'image_classification/imagewang_labels.txt',
+    'image_classification/inaturalist_labels.txt',
+    'image_classification/inaturalist_supercategories.txt',
+    'image_classification/placesfull/categories_placesfull.txt',
+    'image_classification/plantae_k_urls.txt',
+    'image_classification/sun397_labels.txt',
+    'image_classification/sun397_tfds_te.txt',
+    'image_classification/sun397_tfds_tr.txt',
+    'image_classification/sun397_tfds_va.txt',
+    'object_detection/open_images_classes_all.txt',
+    'object_detection/open_images_classes_boxable.txt',
+    'object_detection/open_images_classes_trainable.txt',
+    'video/tao/labels.txt',
+    'video/ucf101_labels.txt',
+    'video/youtube_vis/labels.txt',
+]
+
+# Extra dependencies required by specific datasets
+DATASET_EXTRAS = {
+    # In alphabetical order
+    'aflw2k3d': ['scipy'],
+    'beir': ['apache-beam<2.65.0'],
+    'ble_wind_field': ['gcsfs', 'zarr<3.0.0'],
+    'c4': [
+        'apache-beam<2.65.0',
+        'gcld3',
+        'langdetect',
+        # nltk==3.8.2 is broken: https://github.com/nltk/nltk/issues/3293
+        'nltk==3.8.1',
+        'tldextract',
+    ],
+    'c4_wsrs': ['apache-beam<2.65.0'],
+    'cats_vs_dogs': ['matplotlib'],
+    'colorectal_histology': ['Pillow'],
+    'common_voice': ['pydub'],  # and ffmpeg installed
+    'duke_ultrasound': ['scipy'],
+    'eurosat': ['scikit-image', 'tifffile', 'imagecodecs'],
+    'groove': ['pretty_midi', 'pydub'],
+    'gtzan': ['pydub'],
+    'imagenet2012_corrupted': [
+        # This includes pre-built source; you may need to use an alternative
+        # route to install OpenCV
+        'opencv-python',
+        'scikit-image',
+        'scipy',
+    ],
+    'librispeech': ['pydub'],  # and ffmpeg installed
+    'lsun': ['tensorflow-io[tensorflow]'],
+    # sklearn version required to avoid conflict with librosa from
+    # https://github.com/scikit-learn/scikit-learn/issues/14485
+    # See https://github.com/librosa/librosa/issues/1160
+    'nsynth': ['crepe>=0.0.11', 'librosa', 'scikit-learn==0.20.3'],
+    'ogbg_molpcba': ['pandas', 'networkx'],
+    'pet_finder': ['pandas'],
+    'qm9': ['pandas'],
+    'robonet': ['h5py'],  # and ffmpeg installed
+    # envlogger is not available for Python versions >= 3.11 or non Linux
+    # platforms: https://pypi.org/project/envlogger/#files
+    'locomotion': ['envlogger;python_version<"3.11" and sys_platform=="linux"'],
+    'robosuite_panda_pick_place_can': [
+        'envlogger;python_version<"3.11" and sys_platform=="linux"'
+    ],
+    'smartwatch_gestures': ['pandas'],
+    'speech_commands': ['pydub'],
+    'svhn': ['scipy'],
+    'the300w_lp': ['scipy'],
+    'wake_vision': ['pandas'],
+    'wider_face': ['Pillow'],
+    'wiki_dialog': ['apache-beam<2.65.0'],
+    'wikipedia': ['apache-beam<2.65.0', 'mwparserfromhell', 'mwxml'],
+    'wsc273': ['bs4', 'lxml'],
+    'youtube_vis': ['pycocotools'],
+}
+
+# Those datasets have dependencies which conflict with the rest of TFDS, so
+# running them in an isolated environments.
+ISOLATED_DATASETS = ('nsynth', 'lsun')
+
+# Extra dataset deps are required for the tests
+all_dataset_dependencies = list(
+    itertools.chain.from_iterable(
+        deps
+        for ds_name, deps in DATASET_EXTRAS.items()
+        if ds_name not in ISOLATED_DATASETS
+    )
+)
+
+TESTS_ALL_DEPENDENCIES = TESTS_DEPENDENCIES + all_dataset_dependencies
+# `datasets` needs to be installed separately in Python >= 3.10 due to
+# conflicts between `multiprocess` and `apache-beam` libraries. See
+# https://github.com/uqfoundation/multiprocess/issues/125
+HUGGINGFACE_ALL_DEPENDENCIES = [
+    dep for dep in TESTS_ALL_DEPENDENCIES if not dep.startswith('apache-beam')
+] + ['datasets']
+
+EXTRAS = {
+    'matplotlib': ['matplotlib'],
+    'tensorflow': ['tensorflow>=2.1'],
+    'tf-nightly': ['tf-nightly'],
+    'tensorflow-data-validation': ['tensorflow-data-validation'],
+    'tests-all': TESTS_ALL_DEPENDENCIES,
+    'dev': TESTS_DEPENDENCIES + DEV_DEPENDENCIES,
+    'huggingface': HUGGINGFACE_ALL_DEPENDENCIES,
+}
+EXTRAS.update(DATASET_EXTRAS)
+
+setuptools.setup(
+    name=project_name,
+    version=__version__,
+    description=DOCLINES[0],
+    long_description='\n'.join(DOCLINES[2:]),
+    author='Google Inc.',
+    author_email='packages@tensorflow.org',
+    url='https://github.com/tensorflow/datasets',
+    download_url='https://github.com/tensorflow/datasets/tags',
+    license='Apache 2.0',
+    packages=setuptools.find_packages(),
+    package_data={
+        'tensorflow_datasets': DATASET_FILES + [
+            # Bundle `datasets/` folder in PyPI releases
+            'datasets/*/*',
+            'core/utils/colormap.csv',
+            'scripts/documentation/templates/*',
+            'url_checksums/*',
+            'checksums.tsv',
+            'community-datasets.toml',
+            'dataset_collections/*/*.md',
+            'dataset_collections/*/*.bib',
+            'core/valid_tags.txt',
+        ],
+    },
+    exclude_package_data={
+        'tensorflow_datasets': [
+            'dummy_data/*',
+        ],
+    },
+    scripts=[],
+    install_requires=REQUIRED_PKGS,
+    python_requires='>=3.10',
+    extras_require=EXTRAS,
+    classifiers=[
+        'Development Status :: 4 - Beta',
+        'Intended Audience :: Developers',
+        'Intended Audience :: Science/Research',
+        'License :: OSI Approved :: Apache Software License',
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3 :: Only',
+        'Topic :: Scientific/Engineering :: Artificial Intelligence',
+    ],
+    keywords='tensorflow machine learning datasets',
+    entry_points={
+        'console_scripts': [
+            'tfds = tensorflow_datasets.scripts.cli.main:launch_cli'
+        ],
+    },
+    # Include_package_data is required for setup.py to recognize the MANIFEST.in
+    #   https://python-packaging.readthedocs.io/en/latest/non-code-files.html
+    include_package_data=True,
+)

tfds_nightly-4.9.9.dev202506030044/tensorflow_datasets/__init__.py

@@ -0,0 +1,91 @@
+# coding=utf-8
+# Copyright 2025 The TensorFlow Datasets Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# pylint: disable=line-too-long
+"""`tensorflow_datasets` (`tfds`) defines a collection of datasets ready-to-use with TensorFlow.
+
+Each dataset is defined as a `tfds.core.DatasetBuilder`, which encapsulates
+the logic to download the dataset and construct an input pipeline, as well as
+contains the dataset documentation (version, splits, number of examples, etc.).
+
+The main library entrypoints are:
+
+* `tfds.builder`: fetch a `tfds.core.DatasetBuilder` by name
+* `tfds.load`: convenience method to construct a builder, download the data, and
+  create an input pipeline, returning a `tf.data.Dataset`.
+
+Documentation:
+
+* These API docs
+* [Available datasets](https://www.tensorflow.org/datasets/catalog/overview)
+* [Colab
+tutorial](https://colab.research.google.com/github/tensorflow/datasets/blob/master/docs/overview.ipynb)
+* [Add a dataset](https://www.tensorflow.org/datasets/add_dataset)
+"""
+# pylint: enable=line-too-long
+# pylint: disable=g-import-not-at-top,g-bad-import-order,wrong-import-position,unused-import
+
+from __future__ import annotations
+
+from absl import logging
+from etils import epy as _epy
+
+
+try:
+  # pylint: disable=g-import-not-at-top
+  # pytype: disable=import-error
+  # For builds that don't include all dataset builders, we don't want to fail on
+  # import errors of dataset builders.
+  with _epy.lazy_imports(
+      error_callback='Could not import TFDS dataset builders.'
+  ):
+    from tensorflow_datasets import audio
+    from tensorflow_datasets import graphs
+    from tensorflow_datasets import image
+    from tensorflow_datasets import image_classification
+    from tensorflow_datasets import object_detection
+    from tensorflow_datasets import nearest_neighbors
+    from tensorflow_datasets import question_answering
+    from tensorflow_datasets import d4rl
+    from tensorflow_datasets import ranking
+    from tensorflow_datasets import recommendation
+    from tensorflow_datasets import rl_unplugged
+    from tensorflow_datasets.rlds import datasets
+    from tensorflow_datasets import robotics
+    from tensorflow_datasets import robomimic
+    from tensorflow_datasets import structured
+    from tensorflow_datasets import summarization
+    from tensorflow_datasets import text
+    from tensorflow_datasets import text_simplification
+    from tensorflow_datasets import time_series
+    from tensorflow_datasets import translate
+    from tensorflow_datasets import video
+    from tensorflow_datasets import vision_language
+
+  # pytype: enable=import-error
+
+  from tensorflow_datasets import rlds  # pylint: disable=g-bad-import-order
+
+  # Public API to create and generate a dataset
+  from tensorflow_datasets.public_api import *  # pylint: disable=wildcard-import
+  from tensorflow_datasets import public_api  # pylint: disable=g-bad-import-order
+  # pylint: enable=g-import-not-at-top
+  # __all__ for import * as well as documentation
+  __all__ = public_api.__all__
+
+except Exception as exception:  # pylint: disable=broad-except
+  logging.exception(exception)
+
+del _epy