llm-d-benchmark/pyproject.toml at main · llm-d/llm-d-benchmark · 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
[project]
name = "llmdbenchmark"
version = "0.6.0"
description = "A library for configuration discovery and performance benchmarking for llm-d."
requires-python = ">=3.11"
dependencies = [
    "PyYAML",
    "Jinja2",
    "requests",
    "packaging",
    "kubernetes",
    "pykube-ng",
    "kubernetes-asyncio",
    "GitPython",
    "huggingface_hub",
    "transformers",
    "pydantic>=2.0",
    "google-auth",
    "google-cloud-storage>=2.10.0",
]

#
# When building llmdbenchmark, we will exclude anything not immmediately
# part of the library itself, which entails any templates or specification files.
# If we do not do this, we run the risk of setuptools prematurely thinking these
# are part of the overall distribution.
#

[tool.setuptools.packages.find]
where = ["."]
include = ["llmdbenchmark*"]
exclude = ["templates*", "specification*"]

[tool.setuptools.package-data]
"llmdbenchmark.analysis" = [
    "scripts/*.sh",
    "scripts/*.py",
    "benchmark_report/*.yaml",
    "benchmark_report/*.json",
]

[project.urls]
Homepage = "https://llm-d.ai"
Repository = "https://github.com/llm-d/llm-d-benchmark"

[project.scripts]
llmdbenchmark = "llmdbenchmark.cli:cli"