kl-eval/pyproject.toml at main · trymirai/kl-eval · 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
[project]
name = "lalamo-eval"
version = "0.1.0"
description = "Top-k distribution evaluation for Lalamo and llama.cpp checkpoints."
readme = "README.md"
requires-python = ">=3.12,<4"
dependencies = [
    "cattrs>=25.3.0,<26",
    "datasets>=4.8.5,<5",
    "huggingface-hub>=0.27.1",
    "jax>=0.10.0,<0.11",
    "jaxtyping>=0.2.36",
    "lalamo==0.13.1",
    "llama-cpp-python==0.3.23",
    "numpy>=2.0.0,<3",
    "tokenizers>=0.21.2,<0.24",
]

[project.optional-dependencies]
cuda = ["jax[cuda]>=0.10.0,<0.11", "nvidia-cutlass-dsl>=4.4,<4.5; platform_system == 'Linux'"]

[dependency-groups]
dev = [
    "pre-commit>=4.0.0",
    "pytest>=8.0.0",
    "pyrefly>=0.60.2",
    "ruff>=0.14.0",
]

[project.scripts]
lalamo-eval = "lalamo_eval.cli:main"

[build-system]
requires = ["setuptools>=65.5.0", "wheel"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
where = ["."]
include = ["lalamo_eval*"]
namespaces = false

[tool.uv]
package = true
constraint-dependencies = [
    "nvidia-cudnn-cu12==9.21.0.82; platform_system == 'Linux'",
    "polars<1.41",
]

[tool.ruff]
target-version = "py312"
line-length = 119

[tool.ruff.lint]
select = ["ALL"]
ignore = [
    "COM812",
    "D",
    "EM101",
    "EM102",
    "FBT003",
    "PLC0415",
    "PLR0911",
    "PLR0913",
    "PLR2004",
    "S101",
    "T201",
    "TC001",
    "TC002",
    "TC003",
    "TRY003",
    "TRY004",
]

[tool.pyrefly]
python-version = "3.12.0"
infer-with-first-use = false
python-interpreter-path = ".venv/bin/python"
project-includes = ["./"]
ignore-missing-imports = ["lalamo.*"]