LLM-DNA/pyproject.toml at main · Xtra-Computing/LLM-DNA · 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
[project]
name = "llm-dna"
version = "0.2.2"
description = "Extract LLM DNA vectors — low-dimensional representations that capture functional behavior and model evolution."
authors = [{ name = "LLM-DNA Project" }]
license = { file = "LICENSE" }
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: Apache Software License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

dependencies = [
    "numpy>=1.24.0",
    "torch>=2.4.0",
    "transformers>=4.40.0",
    "accelerate>=0.26.0",
    "huggingface-hub>=0.23.0",
    "datasets>=2.14.0",
    "pandas>=2.0.0",
    "sentence-transformers>=2.2.2",
    "scikit-learn>=1.3.0",
    "tqdm>=4.65.0",
    "pyyaml>=6.0",
    "wonderwords>=2.2.0",
    "openai>=1.0.0",
    "tiktoken>=0.7.0",
]

[project.optional-dependencies]
model_scraping = ["requests>=2.31.0"]

vllm = ["vllm>=0.4.0"]

dev = [
    "pytest>=7.4.0",
    "pytest-cov>=4.1.0",
]

[project.scripts]
llm-dna = "llm_dna.cli:main"
calc-dna = "llm_dna.cli:main"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["src/llm_dna"]

[tool.hatch.build.targets.sdist]
include = [
    "src/llm_dna/",
    "scripts/calc_dna.sh",
    "scripts/calc_dna.py",
    "configs/llm_list.txt",
    "README.md",
    "LICENSE",
]

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
addopts = "-q"
markers = [
    "slow: marks tests as slow (run with --run-slow)",
]