pydantic>=2.1.1
torch==2.0.1
confection>=0.1.1
pandas>=2.0.3
lightning>=2.0.9
wandb>=0.15.8
pyarrow>=12.0.0
polars>=0.19.3

[dev]
cruft>=2.0.0
pyright>=1.1.337
pyright-polite>=0.0.1
pre-commit>=2.20.0
ruff>=0.0.254
black[jupyter]>=22.8.0
invoke>=2.2.0

[docs]
jupyter>=1.0.0

[tests]
pytest>=7.1.3
pytest-cov>=3.0.0
pytest-xdist>=3.0.0
pytest-sugar>=0.9.4
tox>=4.5.0
