granite-switch/pyproject.toml at main · primeqa/granite-switch · 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
[build-system]
requires = ["setuptools>=64", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "granite-switch"
version = "0.1.0"
description = "Granite Switch: Composable model building"
readme = "README.md"
license = "Apache-2.0"
requires-python = ">=3.10,<3.14"
dependencies = [
    "torch>=2.10.0",
    "transformers>=5.5.1",
]

[project.urls]
Repository = "https://github.com/generative-computing/granite-switch"
Documentation = "https://github.com/generative-computing/granite-switch/tree/main/tutorials"

[project.optional-dependencies]
hf = ["accelerate>=0.20.0"]
vllm = ["vllm>=0.19.1,<0.20.0"]
vllm20 = ["vllm>=0.20.0,<0.21.0"]
compose = ["huggingface_hub", "pyyaml", "tqdm", "safetensors"]
build = ["huggingface_hub", "pyyaml", "tqdm", "safetensors"]  # Backward compatibility alias for compose
tutorials = [
    "granite-switch[hf,vllm,compose]",
    "chromadb>=0.4.0",
    "httpx>=0.24.0",
    "requests>=2.31.0",
    "rich>=13.0.0",
    "mellea>=0.1.0,<=0.5.0",
    "ipython>=8.10.0",
    "python-dotenv>=1.0.0",
]
dev = ["pytest", "granite-switch[hf,vllm,compose]"]
dev-vllm20 = ["pytest", "granite-switch[hf,vllm20,compose]"]
test = ["pytest", "bitsandbytes", "optimum-quanto"]

[project.entry-points."vllm.general_plugins"]
register_granite_switch = "granite_switch.vllm:register"

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = ["--ignore=tests/_legacy", "-m", "not deep"]
markers = [
    "gpu: requires CUDA GPU",
    "vllm: requires vLLM installed",
    "slow: takes > 30s",
    "deep: expensive code-theory tests (m=8 / 256-dim); run with: pytest -m deep",
    "requires_model: needs a real model checkpoint",
]

[tool.uv]
conflicts = [
    [
        { extra = "vllm" },
        { extra = "vllm20" },
    ],
    [
        { extra = "dev" },
        { extra = "vllm20" },
    ],
    [
        { extra = "dev" },
        { extra = "dev-vllm20" },
    ],
    [
        { extra = "dev-vllm20" },
        { extra = "vllm" },
    ],
    [
        { extra = "tutorials" },
        { extra = "vllm20" },
    ],
    [
        { extra = "tutorials" },
        { extra = "dev-vllm20" },
    ],
]

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