AI-trainer/pyproject.toml at develop · K-BEL/AI-trainer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "siin-trainer"
version = "0.1.0"
description = "A generic Python library to train AI models on custom datasets."
authors = [
    { name="Siin Lab", email="contact@siinlab.com" }
]
license = { text = "MIT" }
readme = "README.md"
requires-python = ">=3.11,<3.13"
keywords = ["AI", "machine learning", "model training", "custom datasets"]
dependencies = [
    "click",
    "lgg",
    "tqdm",
    "pyyaml",
    "requests",
    "numpy",
    "ultralytics",
    "wandb>=0.21.0",
    "rfdetr[metrics]>=1.2.1",
    "clip",
    "rich",
    "streamlit",
]

[project.optional-dependencies]
dev = ["ruff"]

[project.scripts]
siin-trainer = "trainer.cli:main"

[tool.setuptools.packages.find]
where = ["."]

[tool.uv.sources]
clip = { git = "https://github.com/openai/CLIP.git", rev = "d05afc436d78f1c48dc0dbf8e5980a9d471f35f6" }