cube-standard/pyproject.toml at dev · The-AI-Alliance/cube-standard · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
[project]
name = "cube-standard"
version = "0.1.0rc11"
description = "Common Unified Benchmark Environments"
readme = "README.md"
urls = { Repository = "https://github.com/The-AI-Alliance/cube-standard" }
authors = [
    { name = "Nicolas Gontier", email = "nicolas.gontier@servicenow.com" },
    { name = "Aman Jaiswal", email = "amanjaiswal73892@gmail.com" },
    { name = "Alexandre Lacoste", email = "alexandre.lacoste@servicenow.com" },
]
requires-python = ">=3.12"
dependencies = [
    "fastapi>=0.115.0",
    "docstring_parser>=0.16",
    "pillow>=9.0",
    "pydantic>=2.0",
    "pydantic-settings>=2.8.0",
    "python-dotenv>=1.2.1",
    "requests>=2.28",
    "rich>=13.0",
    "tqdm>=4.60",
    "typing-extensions>=4.12",
    "uvicorn[standard]>=0.34.0",
]

[project.scripts]
cube = "cube.cli:main"

[project.optional-dependencies]
dev = [
    "pytest>=8.0.0",
    "pytest-asyncio>=0.24.0",
    "pytest-xdist>=3.0.0",
    "httpx>=0.28.0",  # For testing FastAPI
    "pre-commit>=3.6.0",
    "ruff>=0.8.0",
    "counter-cube",  # example benchmark → registers cube.benchmarks entry point for `cube test counter-cube`
]
docker = [
    "docker>=7.1.0",
    "tenacity>=9.1.4",
]
daytona = [
    "daytona>=0.142.0",
    "tenacity>=9.1.4",
]
modal = [
    "modal>=1.3.3",
    "tenacity>=9.1.4",
]
toolkit = [
    "tenacity>=9.1.4",
]

[build-system]
requires = ["uv_build>=0.6.0,<0.7.0"]
build-backend = "uv_build"

[tool.uv.workspace]
members = ["cube-resources/*"]

[tool.uv.sources]
counter-cube = { path = "examples/counter-cube", editable = true }

[tool.uv.build-backend]
module-name = "cube"
include = ["src/cube/scripts/install_local_infra.sh"]

[tool.ruff]
fix = true
line-length = 120
indent-width = 4
# CI/utility scripts are not held to the library's lint/format style.
extend-exclude = [".github/scripts"]

[tool.ruff.format]
quote-style = "double"
indent-style = "space"
skip-magic-trailing-comma = false
line-ending = "auto"

[tool.ruff.lint]
extend-select = ["I"] # sort imports

[tool.ruff.lint.per-file-ignores]
# Integration tests use pytest.importorskip / pytest.skip before imports
"tests/test_local.py" = ["E402"]
"tests/test_modal.py" = ["E402"]
"tests/test_daytona.py" = ["E402"]
"tests/test_toolkit.py" = ["E402"]

[tool.pytest.ini_options]
testpaths = ["tests", "cube-resources"]
addopts = "-rs -m 'not integration'"
asyncio_mode = "auto"
markers = [
    "integration: end-to-end tests that hit GitHub (run with -m integration)",
]

[dependency-groups]
dev = [
    "pre-commit>=4.5.1",
]