forked from OpenPipe/ART
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpyproject.toml
More file actions
65 lines (55 loc) · 1.58 KB
/
pyproject.toml
File metadata and controls
65 lines (55 loc) · 1.58 KB
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
[project]
name = "openpipe-art"
version = "0.4.4"
description = "The OpenPipe Agent Reinforcement Training (ART) library"
readme = "README.md"
requires-python = ">=3.10"
dependencies = ["openai>=1.65.5", "typer>=0.15.2", "litellm>=1.63.0","weave>=0.51.51",]
[project.optional-dependencies]
plotting = ["matplotlib>=3.10.1", "seaborn>=0.13.2"]
backend = [
"peft>=0.14.0",
"hf-xet>=1.1.0",
"bitsandbytes>=0.45.2",
"unsloth==2025.6.12",
"unsloth-zoo==2025.6.8",
"vllm==0.9.1",
"torchtune",
"trl>=0.19.0",
"torch>=2.7.0",
"torchao>=0.9.0",
"accelerate==1.7.0",
"awscli>=1.38.1",
"setproctitle>=1.3.6",
"tblib>=3.0.0",
"setuptools>=78.1.0",
"wandb>=0.19.8",
"polars>=1.26.0",
]
skypilot = ["semver>=3.0.4", "skypilot[cudo,do,fluidstack,gcp,lambda,paperspace,runpod]==0.9.3"]
[project.scripts]
art = "art.cli:app"
stop-server = "art.skypilot.stop_server:main"
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[tool.hatch.metadata]
allow-direct-references = true
[tool.hatch.build.targets.wheel]
packages = ["src/art", "src/mp_actors"]
[tool.hatch.build]
sources = ["src"]
[tool.uv]
required-version = ">=0.6.15"
dev-dependencies = [
"black>=25.1.0",
"ipykernel>=6.29.5",
"ipywidgets>=8.1.5",
"openpipe>=4.49.0",
"hatch>=1.14.1",
"ruff>=0.12.1",
]
[tool.uv.sources]
panza = { git = "https://github.com/corbt/panza.git" }
torchtune = { git = "https://github.com/pytorch/torchtune.git", rev = "2344509cf83bd886538fe3e8263e5145d1afb5c2" }
sweagent = { git = "https://github.com/bradhilton/SWE-agent" }