datasets
fpdf
pandas
numpy
matplotlib
scikit-learn
tensorflow<2.16,>=2.13
tensorflow_probability==0.22
pyarrow
seaborn

[dev]
pytest>=3.9
pytest-cov
black
twine
setuptools
wheel
pylint

[wandb]
wandb>=0.15
