ATLAS/pyproject.toml at main · Arc-Computer/ATLAS · 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
[build-system]
requires = ["setuptools>=77.0.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "atlas_core"
version = "0.1.0"
description = "Atlas Core: offline training engine for adaptive agents."
readme = { file = "README.md", content-type = "text/markdown" }
license = { text = "MIT" }
authors = [{ name = "Arc Computer" }]
requires-python = ">=3.11"
keywords = ["rlhf", "grpo", "gkd", "sft", "agents", "training"]
classifiers = [
  "Development Status :: 3 - Alpha",
  "Intended Audience :: Developers",
  "Intended Audience :: Science/Research",
  "License :: OSI Approved :: MIT License",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
]
dependencies = [
  "accelerate>=1.4.0",
  "datasets>=3.2.0",
  "hf-transfer>=0.1.9",
  "hydra-core>=1.3.2",
  "litellm>=1.77.7",
  "numpy",
  "packaging",
  "pyyaml",
  "tensorboard>=2.18.0",
  "torch>=2.1.0",
  "transformers>=4.51.1",
  "trl>=0.14.0",
  "wandb",
]

[project.optional-dependencies]
deepspeed = ["deepspeed>=0.15.4"]
peft = ["peft>=0.14.0"]
ray = ["ray[serve]>=2.43.0"]
quantization = [
  "bitsandbytes==0.45.5; platform_system == \"Linux\" and platform_machine != \"aarch64\"",
  "bitsandbytes==0.48.2; platform_system == \"Linux\" and platform_machine == \"aarch64\"",
]
vllm = ["vllm", "fastapi", "pydantic", "requests", "uvicorn"]
sdk = ["arc-atlas>=0.1.15"]
dev = [
  "pre-commit",
  "pytest",
  "pytest-cov",
  "pyright",
  "ruff",
]

[project.scripts]
atlas-core = "atlas_core.cli.main:main"

[tool.setuptools]
package-dir = { "" = "src" }

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

[tool.setuptools.package-data]
atlas_core = ["configs/**/*.yaml", "configs/**/*.md", "configs/**/*.json"]