polars
pyarrow
nested_ragged_tensors>=0.1.0
loguru
numpy
ml-mixins
torch
omegaconf
hydra-core
lightning
x-transformers==1.34.0
rich
hydra-colorlog
hydra-optuna-sweeper
wandb
torchmetrics
torchvision
transformers
ray[tune]
clinical-zeroshot-labeler==0.0.3
lightning-bolts

[dev]
pre-commit<4
notebook
pdbpp

[docs]
mkdocs==1.6.0
mkdocs-gen-files==0.5.0
mkdocs-get-deps==0.2.0
mkdocs-git-authors-plugin==0.9.0
mkdocs-git-revision-date-localized-plugin==1.2.7
mkdocs-literate-nav==0.6.1
mkdocs-material==9.5.33
mkdocs-material-extensions==1.3.1
mkdocs-section-index==0.3.9
mkdocs-snippets==1.3.0
mkdocstrings==0.25.2
mkdocstrings-python==1.10.8

[tests]
pytest
pytest-cov[toml]
rootutils
pytest-xdist
sh
MEDS-transforms>=0.0.9
pytest-loguru
