torch>=1.9.0
torchvision>=0.10.0
numpy>=1.19.2
tqdm>=4.62.3
scikit-learn>=0.24.2
pillow>=8.3.1
matplotlib>=3.4.3
transformers>=4.0.0
medmnist>=2.0.0

[all]
pytest>=6.2.5
pytest-cov
black
isort
flake8
wandb>=0.12.0
eco2ai>=1.0.0

[dev]
pytest>=6.2.5
pytest-cov
black
isort
flake8

[tracking]
wandb>=0.12.0
eco2ai>=1.0.0
