pyproject.toml

[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "lego-rl"
version = "0.1.0"
description = "Integration patches for agentic RL training with VeRL and Harbor."
readme = "README.md"
requires-python = ">=3.12,<3.14"
license = "Apache-2.0"
license-files = ["LICENSE"]
authors = [
  { name = "Lego-RL maintainers" }
]
keywords = ["agentic-rl", "verl", "harbor", "ppo", "grpo", "kubernetes"]
dependencies = [
  "harbor>=0.3.1",
  "numpy>=1.26,<3",
  "omegaconf>=2.3",
  "pandas>=2.2.2",
  "pyarrow>=19.0.0",
]

[project.optional-dependencies]
train = [
  "accelerate",
  "codetiming",
  "compressed-tensors==0.18.0",
  "datasets",
  "dill",
  "hydra-core",
  "peft",
  "pybind11",
  "pylatexenc",
  "ray[default]>=2.41.0",
  "torch==2.10.0",
  "torchaudio==2.10.0",
  "torchdata",
  "torchvision==0.25.0",
  "tensordict>=0.8.0,<=0.10.0,!=0.9.0",
  "transformers==5.4.0",
  "vllm==0.19.0",
  "verl",
  # Hard requirement of the default modeling backend: verl's
  # workers/engine/veomni/transformer_impl.py imports veomni at module load and
  # MODELING_BACKEND defaults to veomni. Its stale `datasets<=2.21.0` pin is lifted by
  # the datasets override below -- installing this extra without that override in
  # effect will fail to resolve. See overrides.txt.
  "veomni==0.1.11",
  "blobfile>=3.0.0",
  "timm",
  "wandb",
  "packaging>=20.0",
  "tensorboard",
]
agent = [
  "openhands-ai",
  "openhands-sdk",
  "openhands-tools",
]
cuda12x = [
  "cupy-cuda12x",
]
dev = [
  "pytest>=8.4.2",
  "pytest-asyncio>=1.2.0",
  "ruff>=0.15.4",
]

[tool.uv]
# Lifts stale upstream metadata so `[train]` resolves. Keep these exact pins in
# sync with requirements.txt and overrides.txt.
#
# Caveat: uv only picks this up when it discovers this pyproject.toml as its config
# file, i.e. when the working directory is the repo root. scripts/setup_env.sh does
# not rely on that -- it exports UV_OVERRIDE=<repo>/overrides.txt so the override
# applies no matter where the script is invoked from. This block is here so that a
# plain `uv sync` / `uv pip install -e .[train]` run from the repo root also works.
override-dependencies = [
  "datasets==4.8.4",
  "transformers==5.4.0",
  "compressed-tensors==0.18.0",
]

[tool.setuptools]
include-package-data = true

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

[tool.setuptools.package-data]
"verl_patch" = ["config/*.yaml"]
"harbor_patch" = ["**/*.j2"]