torch>=1.10
pytorch-lightning>=1.5
pyyaml
pandas
scikit-learn
wfdb
matplotlib
tensorboard

[all]
huggingface_hub>=0.20
python-dotenv>=1.0
cairosvg>=2.5
transformers>=4.30
neurokit2>=0.2

[dev]
pytest
black
flake8
mypy

[eval]
neurokit2>=0.2

[hf]
huggingface_hub>=0.20
python-dotenv>=1.0

[plot]
cairosvg>=2.5

[ssm]
mamba-ssm>=2.0

[text]
transformers>=4.30
