pyproject.toml

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

[project]
name = "yeto"
version = "0.1.0"
description = "Efficient, low-cost post-training across clouds and regions"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT" }
dependencies = [
    # <2.9: cu13x wheels need driver >=580; cloud AMIs ship 535 (CUDA 12.2).
    "torch>=2.4,<2.9",
    "transformers==5.13.0",  # keep in lockstep with requirements.txt
    "datasets>=3.0",
    "peft>=0.14",
    "accelerate>=1.0",
    "sentencepiece",
    "protobuf",
    "cloudpickle",
    "huggingface-hub>=0.34",
    "safetensors>=0.4",
]

[project.optional-dependencies]
launcher = ["skypilot[aws]>=0.12"]
# Apple-silicon island backend (yeto.mlx.learner); torch stays required —
# the wire/data path is shared with the CUDA learners.
mlx = ["mlx>=0.31", "mlx-lm>=0.29"]
qlora = ["bitsandbytes>=0.46.1"]
local-rl = ["gymnasium>=0.29.0", "requests>=2.31.0", "numpy>=1.24"]
diffusion = ["diffusers>=0.35", "safetensors", "pillow", "imageio[ffmpeg]", "bitsandbytes>=0.46.1"]
diffusion-protenix = ["protenix; python_version >= '3.11'"]
a100-liger = ["liger-kernel==0.8.0", "peft==0.19.1"]
a100-flash-attn = ["flash-attn==2.8.3"]
wandb = ["wandb>=0.18"]
dev = ["pytest", "pytest-timeout"]

[project.scripts]
yeto = "yeto.cli:main"
yeto-export = "yeto.export:main"
yeto-codex-traces = "yeto.codex_traces:main"
yeto-wandb-tape = "yeto.wandb_tape:main"
yeto-diffusion-export = "yeto.diffusion.export:main"
yeto-diffusion-sample = "yeto.diffusion.sample:main"
yeto-rl-export = "yeto.rl.export:main"
yeto-rl-ssh = "yeto.rl.ssh_harness:main"
yeto-protenix-export-batch = "yeto.diffusion.protenix_export:main"
yeto-hunyuan3d-export-batch = "yeto.diffusion.hunyuan3d_export:main"

[tool.setuptools.packages.find]
include = ["yeto*"]