datasets>=4.0.0
huggingface_hub
fpdf
pandas
numpy
matplotlib
scikit-learn
pyarrow
seaborn

[dev]
pytest>=7.0.0
pytest-cov
black
twine
setuptools
wheel
pylint
tensorflow<2.16,>=2.13
torch
torchvision

[lightning]
lightning

[pytorch]
torch
torchvision

[tensorflow]
tensorflow<2.16,>=2.13

[tf]
tensorflow<2.16,>=2.13

[torch]
torch
torchvision

[wandb]
wandb>=0.20.0
