dill
pandas
psutil
requests
rich
termcolor
tqdm

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

[boosting]
catboost>=1.2
lightgbm>=4.0
xgboost>=1.7

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

[huggingface]
transformers>=4.30
peft>=0.5

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

[sklearn]
numpy
scikit-learn>=1.2

[survival]
lifelines>=0.27
scikit-survival>=0.22

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