pandas>=2.0.1
numpy>=2.1
scikit-learn>=0.21.2
torch>=2.5.1
torchvision>=0.20.1
packaging
pydantic>=2.5.0
torchmetrics>=1.1.2

[all]
neuraltrain[lightning,metrics,models]

[dev]
pytest>=4.3.0
pytest-mock
pytest-markdown-docs
ruff>=0.15
pre-commit>=1.15.2
mypy>=0.971
pandas-stubs
types-PyYAML
types-Pillow
types-pywin32
types-requests
types-setuptools
types-tabulate
types-tqdm

[lightning]
lightning<=2.6.1,>=2.0.8
wandb>=0.15.11

[metrics]
openai-clip

[models]
braindecode>=1.4.0
x_transformers>=1.27.20
dalle2_pytorch
vit_pytorch
