TuFT/pyproject.toml at 2409ef32869c048e9d7596d3bfafd3edd783b688 · agentscope-ai/TuFT · 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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
[project]
name = "tuft"
version = "0.1.3"
description = "A multi-tenant fine-tuning platform for LLMs with Tinker-compatible API"
authors = [
    { name = "TuFT Developers", email = "tuft@list.alibaba-inc.com" }
]
readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.11"
dependencies = [
    "fastapi>=0.125.0",
    "httpx>=0.28.1",
    "numpy<2.0.0",
    "tinker>=0.7.0",
    "typer>=0.20.1",
    "uvicorn[standard]>=0.38.0",
    "omegaconf>=2.3.0",
    "ray>=2.50.0",
    "transformers>=4.57.3,<5.0.0",
    "opentelemetry-api>=1.20.0",
    "opentelemetry-sdk>=1.20.0",
    "opentelemetry-exporter-otlp>=1.20.0",
    "opentelemetry-instrumentation-fastapi>=0.41b0",
    "opentelemetry-instrumentation-logging>=0.41b0",
    "psutil>=5.9.0",
    "nvidia-ml-py>=13.0.0",
]

[project.scripts]
"tuft" = "tuft.cli:main"

[project.optional-dependencies]

dev = [
    "detect-secrets>=1.5.0",
    "nbqa>=1.9.1",
    "pre-commit>=4.5.1",
    "pyright>=1.1.408",
    "pytest>=9.0.2",
    "pytest-asyncio>=1.3.0",
    "pytest-json-ctrf",
    "ruff>=0.14.14",
]
backend = [
    "trinity-rft[vllm]>=0.4.1",
    "peft>=0.18.0",
    "datasets>=4.0.0",
    "huggingface-hub>=0.20.0,<1.0"
]
persistence = [
    "redis>=5.0.0",
]
examples = [
    "matplotlib>=3.8",
]
docs = [
    "sphinx>=7.0.0,<9.0.0",
    "myst-parser>=3.0.0",
    "sphinx-copybutton",
    "sphinx-design",
    "linkify-it-py",
    "sphinx-autobuild",
    "sphinxawesome-theme>=6.0.0",
]

[tool.ruff]
line-length = 100
target-version = "py311"
exclude = [
    "tinker",
]

[tool.ruff.lint]
select = ["E", "F", "B", "I"]
per-file-ignores = {"src/tuft/server.py" = ["B008"], "tests/test_fsdp_*.py" = ["E402", "I001"]}

[tool.ruff.lint.isort]
known-first-party = ["tuft"]
known-third-party = ["tinker"]
lines-after-imports = 2
combine-as-imports = true
split-on-trailing-comma = true

[tool.ruff.format]
docstring-code-format = true
quote-style = "double"

[tool.pyright]
pythonVersion = "3.11"
typeCheckingMode = "standard"
reportUnusedImport = false
reportMissingImports = false
exclude = [
    "**/.venv",
    "**/__pycache__",
    ".pytest_cache",
    ".ruff_cache",
    "tinker",
]

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = "test_*.py"
python_classes = "Test*"
python_functions = "test_*"
addopts = "-q"
asyncio_mode = "auto"
markers = [
    "integration: marks tests as integration tests (deselect with '-m \"not integration\"')",
]

[tool.nbqa.addopts]
ruff = ["--fix"]

[tool.nbqa.exclude]
ruff = "^(\\.venv|tinker)/"

[tool.uv.extra-build-dependencies]
flash-attn = ["torch", "numpy"]

[build-system]
requires = ["hatchling", "setuptools>=65", "wheel"]
build-backend = "hatchling.build"