pytest
polars<2,>=1.30
nested_ragged_tensors<0.4,>=0.3
numpy<3,>=1.26
torch>=2.0
meds<0.5,>=0.4
MEDS_transforms<0.7,>=0.6.7
hydra-core>=1.3
omegaconf>=2.3
meds_testing_helpers<0.4,>=0.3
yaml_to_disk[parquet]

[docs]
mkdocs==1.6.1
mkdocs-material==9.6.7
mkdocstrings[python]==0.28.2
mkdocstrings-shell~=1.0
mkdocs-gen-files==0.5.0
mkdocs-literate-nav==0.6.1
mkdocs-section-index==0.3.9
mkdocs-git-authors-plugin==0.9.2
mkdocs-git-revision-date-localized-plugin==1.3.0
markdown-callouts

[joblib]
hydra-joblib-launcher~=1.2.0

[lightning]
lightning~=2.5.1
