state-model-interface/pyproject.toml at master · aabbdev/state-model-interface · 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
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "state-model-interface"
version = "0.1.0"
description = "Secure token-ID compiler for the State Model Interface"
readme = "README.md"
requires-python = ">=3.10"

[project.optional-dependencies]
train = [
    "datasets>=5,<6",
    "flash-linear-attention[tilelang] @ git+https://github.com/fla-org/flash-linear-attention.git@27967b970eaaf982a6960abf6cba8add9c34c7cc ; platform_system == 'Linux'",
    "numba>=0.67,<0.68 ; platform_system == 'Linux'",
    "numpy>=1.26,<2.5",
    "tensorboard>=2.20,<3",
    "torch>=2.8,<2.9",
    "transformers>=5.15,<6",
    "trl>=1.10,<1.11",
]

[project.scripts]
smi-prepare-pilot = "state_model_interface.prepare_pilot:main"
smi-train-rwkv7 = "state_model_interface.train_rwkv7:main"

[dependency-groups]
dev = [
    "datasets>=5,<6",
    "numpy>=1.26,<2.5",
    "pyright>=1.1.403",
    "pytest>=8.0",
    "ruff>=0.9",
    "tensorboard>=2.20,<3",
    "torch>=2.8,<2.9",
    "transformers>=5.15,<6",
    "trl>=1.10,<1.11",
]

[tool.pytest.ini_options]
pythonpath = ["src"]
testpaths = ["tests"]

[tool.ruff]
extend-exclude = ["README.md"]
target-version = "py310"

[tool.ruff.lint]
extend-select = ["I", "UP"]

[tool.pyright]
include = ["src", "tests"]
pythonVersion = "3.10"

[tool.hatch.build.targets.wheel.force-include]
"chat_template.jinja" = "state_model_interface/chat_template.jinja"

[tool.hatch.metadata]
allow-direct-references = true