numpy<3,>=1.26.4
torch>=2.4.1
pandas<3,>=2.2.2
scikit-learn<2,>=1.5
xgboost<4,>=3.1
joblib
matplotlib>=3.9
seaborn>=0.13.2
tableone>=0.9.1
tqdm
typer
python-dotenv
pyyaml
optuna>=3.6.1

[dev]
prism-iml[notebooks]
black
flake8
isort
pre-commit
build
twine

[gpu]
cupy-cuda12x<15,>=13.3

[gpu-cuda13x]
cupy-cuda13x<15,>=13.3

[notebooks]
nbconvert>=6.0
nbformat>=5.1
jupyter
jupyterlab
nbautoexport

[openml]
openml

[test]
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-mock>=3.10.0
hypothesis>=6.0.0
