File size: 393 Bytes
e6c164f 416f658 e6c164f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | [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",
]
|