pyyaml>=6.0
pydantic>=2.0.0
pydantic-settings>=2.0.0
sqlalchemy>=2.0.0
sqlmodel>=0.0.14
typer[all]>=0.9.0
rich>=13.0.0
platformdirs>=3.0.0

[all]
kladml[dev,tracking,train]

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.1.0
build>=1.0.0
twine>=4.0.0

[tracking]
mlflow<4.0.0,>=3.0.0

[train]
torch>=2.0.0
numpy>=1.21.0
onnx>=1.14.0
onnxruntime>=1.15.0
h5py>=3.0.0
onnxscript>=0.1.0
