polars~=1.26.0
pyarrow
hydra-core
numpy
meds~=0.4.0
filelock
pytest
meds_testing_helpers~=0.3.0
MEDS-transforms~=0.5.0
meds-torch-data[lightning]~=0.6.2
transformers
torch
torchmetrics
lightning~=2.5.1

[dev]
pre-commit<4
ruff

[mlflow]
mlflow
psutil
pynvml

[tests]
pytest-cov
pretty_print_directory>=0.1.2
yaml_to_disk

[wandb]
wandb
