dill
pandas
psutil
requests
rich
termcolor
tqdm

[all]
tracebloc[boosting,huggingface,pytorch,sklearn,survival,tensorflow]

[boosting]
tracebloc[catboost,lightgbm,xgboost]

[catboost]
catboost>=1.2

[dev]
pytest>=7.0
pytest-cov>=4.0
mypy>=1.5
ruff>=0.1

[huggingface]
transformers>=4.30
peft>=0.5

[lifelines]
lifelines>=0.27

[lightgbm]
lightgbm>=4.0

[pytorch]
numpy
Pillow
torch>=2.0
torchvision>=0.15
torchmetrics>=1.0
timm>=0.9

[scikit-survival]
scikit-survival>=0.22

[sklearn]
numpy
scikit-learn>=1.2

[survival]
tracebloc[lifelines,scikit-survival]

[tensorflow]
tensorflow<3,>=2.10
tensorflow-datasets>=4.8
silence-tensorflow>=1.2

[xgboost]
xgboost>=1.7
