absl-py
dm-tree
immutabledict
numpy
promise
protobuf>=3.20
psutil
pyarrow
requests>=2.19.0
simple_parsing
tensorflow-metadata
termcolor
toml
tqdm
wrapt

[:platform_system == "Linux"]
array_record>=0.5.0

[:python_version < "3.11"]
etils[edc,enp,epath,epy,etree]>=1.6.0

[:python_version < "3.9"]
importlib_resources

[:python_version >= "3.11"]
etils[edc,enp,epath,epy,etree]>=1.9.1

[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.1
tldextract

[c4_wsrs]
apache-beam<2.65.0

[cats_vs_dogs]
matplotlib

[colorectal_histology]
Pillow

[common_voice]
pydub

[dev]
dill
jax[cpu]==0.4.28
jupyter
pytest
pytest-shard
pytest-xdist
apache-beam<2.65.0
conllu
mlcroissant>=1.0.9
pandas
pydub
pyyaml
pylint>=2.6.0
yapf

[duke_ultrasound]
scipy

[eurosat]
scikit-image
tifffile
imagecodecs

[groove]
pretty_midi
pydub

[gtzan]
pydub

[huggingface]
dill
jax[cpu]==0.4.28
jupyter
pytest
pytest-shard
pytest-xdist
conllu
mlcroissant>=1.0.9
pandas
pydub
pyyaml
scipy
gcsfs
zarr<3.0.0
gcld3
langdetect
nltk==3.8.1
tldextract
matplotlib
Pillow
scikit-image
tifffile
imagecodecs
pretty_midi
opencv-python
networkx
h5py
mwparserfromhell
mwxml
bs4
lxml
pycocotools
datasets

[huggingface:python_version < "3.11" and sys_platform == "linux"]
envlogger

[imagenet2012_corrupted]
opencv-python
scikit-image
scipy

[librispeech]
pydub

[locomotion]

[locomotion:python_version < "3.11" and sys_platform == "linux"]
envlogger

[lsun]
tensorflow-io[tensorflow]

[matplotlib]
matplotlib

[nsynth]
crepe>=0.0.11
librosa
scikit-learn==0.20.3

[ogbg_molpcba]
pandas
networkx

[pet_finder]
pandas

[qm9]
pandas

[robonet]
h5py

[robosuite_panda_pick_place_can]

[robosuite_panda_pick_place_can:python_version < "3.11" and sys_platform == "linux"]
envlogger

[smartwatch_gestures]
pandas

[speech_commands]
pydub

[svhn]
scipy

[tensorflow]
tensorflow>=2.1

[tensorflow-data-validation]
tensorflow-data-validation

[tests-all]
dill
jax[cpu]==0.4.28
jupyter
pytest
pytest-shard
pytest-xdist
apache-beam<2.65.0
conllu
mlcroissant>=1.0.9
pandas
pydub
pyyaml
scipy
gcsfs
zarr<3.0.0
gcld3
langdetect
nltk==3.8.1
tldextract
matplotlib
Pillow
scikit-image
tifffile
imagecodecs
pretty_midi
opencv-python
networkx
h5py
mwparserfromhell
mwxml
bs4
lxml
pycocotools

[tests-all:python_version < "3.11" and sys_platform == "linux"]
envlogger

[tf-nightly]
tf-nightly

[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
