lightning
torch<2.1.0,>=1.11.0
torchmetrics<1.1.0,>=0.7.0
numpy<=1.25.3,>=1.17.2
pydantic<2.2.0,>=2.0.0
scikit-learn
colormath
tqdm<4.67.0,>=4.57.0
typer[all]

[:python_version < "3.11"]
toml
