[project]
name = "model"
version = "0.1.0"
description = "Train a KNN Model"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "joblib>=1.5.3",
    "scikit-learn>=1.9.0",
    "sentence-transformers>=5.6.1",
    "transformers>=5.14.1",
]

[tool.poe.tasks]
train = "uv run ./src/train.py"
test = "uv run ./src/test.py"

[dependency-groups]
dev = [
    "poethepoet>=0.46.0",
]