[build-system]
requires = ["setuptools>=69"]
build-backend = "setuptools.build_meta"

[project]
name = "rarelens-ml"
version = "0.1.0"
requires-python = ">=3.12"
# mlflow major must match the API's mlflow-skinny and the tracking server image.
dependencies = ["lightgbm>=4.5", "mlflow>=3,<4", "pandas", "scikit-learn"]

[project.optional-dependencies]
gpu = ["torch"]   # for the optional deep-learning baseline on GPU
db = ["sqlalchemy>=2", "psycopg[binary]>=3"]   # scripts/load-hpo.py writes the HPO tables
dev = ["pytest>=8"]

[tool.setuptools.packages.find]
include = ["rarelens_ml*"]

[tool.pytest.ini_options]
pythonpath = ["."]
testpaths = ["tests"]