PAWN/pyproject.toml at main · thomas-schweich/PAWN · 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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
[project]
name = "pawn"
version = "0.1.0"
description = "PAWN: Playstyle-Agnostic World-model Network for Chess"
requires-python = ">=3.10,<3.13"
license = "Apache-2.0"
dependencies = [
    "chess-engine",
    "numpy~=2.2.0",
    "psutil>=5.9.0",
    "safetensors>=0.4.0",
    "tqdm~=4.67.0",
    "wandb~=0.25.0",
    "huggingface-hub>=0.20.0",
    # Eval + probes
    "seaborn>=0.13.2",
    "matplotlib>=3.10.8",
    "polars>=1.39.0",
    "pyarrow>=23.0.1",
    "zstandard>=0.22.0",  # Lichess PGN extraction
    # Dashboard
    "solara>=1.0.0",
    "plotly>=5.18.0",
    "pandas>=2.0.0",
    "anywidget>=0.9.21",
    # Sweeps
    "optuna>=4.0.0",
    "optuna-dashboard>=0.17.0",
    # Lab manager MCP server
    "fastmcp>=2.0.0",
    "starlette<1.0.0",  # solara incompatible with starlette 1.0
    # Dev
    "pyright>=1.1.0",
    "pytest~=9.0.0",
    "pytest-mock>=3.12.0",
    "hypothesis>=6.100.0",
    "syrupy>=4.7.0",
    "ipykernel>=7.2.0",
]

[project.optional-dependencies]
rocm = ["torch~=2.10.0", "triton-rocm>=3.6.0", "triton>=3.6.0"]
cu128 = ["torch~=2.10.0"]

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

[tool.hatch.build.targets.wheel]
packages = ["pawn"]

[tool.uv.workspace]
members = ["engine"]

[tool.uv]
conflicts = [
    [
        { extra = "rocm" },
        { extra = "cu128" },
    ],
]
environments = ["sys_platform == 'linux'"]

[tool.uv.sources]
chess-engine = { workspace = true }
torch = [
    { index = "pytorch-rocm", extra = "rocm" },
    { index = "pytorch-cu128", extra = "cu128" },
]
triton-rocm = [
    { index = "pytorch-rocm", extra = "rocm" },
]

[[tool.uv.index]]
name = "pytorch-rocm"
url = "https://download.pytorch.org/whl/rocm7.1"
explicit = true

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true


[tool.pyright]
pythonVersion = "3.10"
typeCheckingMode = "basic"
reportMissingTypeStubs = false
reportPrivateImportUsage = false
reportMissingImports = "error"
include = ["pawn"]
exclude = ["pawn/dashboard"]

[tool.pytest.ini_options]
testpaths = ["tests"]
markers = [
    "unit: pure-function / no-I/O tests",
    "integration: multi-module flows, may touch filesystem",
    "smoke: end-to-end sanity checks (import + minimal execution)",
    "slow: tests that take >5s (excluded from default CI)",
    "gpu: tests that require a CUDA/ROCm GPU",
    "known_bug: tests that document a currently-broken behavior (paired with xfail)",
]