torch>=1.11.0
numpy<=3.0.0
pandas<3.0.0,>=1.1.5
scikit-learn<2.0,>=1.3.0
pytorch-lightning<2.7.0,>=2.0.0
scipy<2.0,>=1.8
omegaconf>=2.3.0
torchmetrics<1.9.0,>=0.10.0
einops<0.9.0,>=0.6.0
rich>=11.0.0
scikit-base

[dependencies_2024]
torch==2.5.0
numpy==2.2.0
pandas==2.2.3
scikit-learn==1.6.0
pytorch-lightning==2.4.0
scipy==1.14.1
omegaconf==2.3.0
torchmetrics==1.5.2
einops==0.8.0

[dev]
wget
bump2version==1.0.1
mkdocs-material<9.8,>=9.5
ipython[notebook]<9.0,>8.0
mkdocstrings[python]==0.29.*
mknotebooks==0.8.*
pytest>=5.3.2
pytest-runner>=5.1
torch_optimizer

[extra]
wandb<0.19.0,>=0.15.0
plotly<5.25.0,>=5.13.0
kaleido<0.3.0,>=0.2.0
captum<0.8.0,>=0.5.0
pytorch-tabnet<4.2

[notebooks]
ipywidgets
matplotlib>3.1
