-
Notifications
You must be signed in to change notification settings - Fork 7
Expand file tree
/
Copy pathpyproject.toml
More file actions
90 lines (80 loc) · 2.91 KB
/
pyproject.toml
File metadata and controls
90 lines (80 loc) · 2.91 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
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
[project]
name = "src"
version = "0.0.1"
readme = "README.md"
classifiers = [
"Development Status :: 3 - Alpha",
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
]
requires-python = ">=3.13"
license = { "text" = "MIT" }
dependencies = [
"skyrl-train",
"transformers==4.57.3",
"openhands-tools",
"openhands-agent-server",
"openhands-workspace",
"vllm==0.11.0",
"verifiers>=0.1.6.post0",
"datasets>=4.0.0",
"ipykernel>=7.1.0",
"ipywidgets>=8.1.8",
"matplotlib>=3.10.7",
"seaborn>=0.13.2",
"gcsfs>=2025.3.0",
"lmcache",
# "flashinfer-python",
# "flashinfer-jit-cache",
]
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"
[tool.setuptools]
py-modules = ["swe_grep_oss_env"]
[tool.setuptools.packages.find]
where = ["."]
include = ["src", "software-agent-sdk"]
[dependency-groups]
dev = [
"pytest>=9.0.2",
"pre-commit>=4.3.0",
"psutil>=7.0.0",
"pyright>=1.1.405",
"ruff>=0.12.10",
"pycodestyle>=2.12.0",
]
[tool.uv]
override-dependencies = [
"flash-attn",
# "anthropic>=0.72.1",
# "openai>=2.8.0",
# "torch==2.8.0",
# "torchvision==0.23.0",
# "torchaudio==2.8.0",
]
extra-build-variables = { flash-attn = { FLASH_ATTENTION_SKIP_CUDA_BUILD = "TRUE" } }
[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true
[[tool.uv.index]]
name = "flashinfer-cu128"
url = "https://flashinfer.ai/whl/cu128"
explicit = true
[tool.uv.extra-build-dependencies]
flash-attn = ["torch"]
[tool.uv.sources]
skyrl-train = { git = "https://github.com/adityasoni9998/SkyRL.git", rev = "81e5a97c7430503c0c4e6508497cc5aa01a0c624", subdirectory = "skyrl-train" }
flash-attn = {url = "https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiTRUE-cp313-cp313-linux_x86_64.whl"}
openhands-sdk = { git = "https://github.com/OpenHands/software-agent-sdk.git", rev = "85ecfd9333d2d2cc4404dd460fd38868d9b978e2", subdirectory = "openhands-sdk" }
openhands-tools = { git = "https://github.com/OpenHands/software-agent-sdk.git", rev = "85ecfd9333d2d2cc4404dd460fd38868d9b978e2", subdirectory = "openhands-tools" }
openhands-workspace = { git = "https://github.com/OpenHands/software-agent-sdk.git", rev = "85ecfd9333d2d2cc4404dd460fd38868d9b978e2", subdirectory = "openhands-workspace" }
openhands-agent-server = { git = "https://github.com/OpenHands/software-agent-sdk.git", rev = "85ecfd9333d2d2cc4404dd460fd38868d9b978e2", subdirectory = "openhands-agent-server" }
torch = { index = "pytorch-cu128" }
torchvision = { index = "pytorch-cu128" }
flashinfer-jit-cache = { index = "flashinfer-cu128" }
# flashinfer-python = [
# { url = "https://download.pytorch.org/whl/cu128/flashinfer/flashinfer_python-0.2.6.post1%2Bcu128torch2.7-cp39-abi3-linux_x86_64.whl" }
# ]