CPU-Inference-Experiments/pyproject.toml at main · JeffWigger/CPU-Inference-Experiments · 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
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "cpu-inference-workspace"
version = "0.0.1"
description = "CPU inference workspace optimized for Intel"
readme = "README.md"
authors = [{ name = "Jeffrey Wigger" }]
requires-python = ">=3.12"
license = { text = "MIT" }
dependencies = [
    "torch",
    "torchvision",
    "torchao>=0.9.0",
    "numpy<2.3.0",
    "jupyterlab>=4.2.3",
    "onnxscript>=0.4.0",
    "onnxruntime>=1.22.0",
    "fastapi>=0.115.3",
    "httpx>=0.27.0",
    "fast-dynamic-batcher>=1.0.1",
    "openvino==2025.2.0",
    "intel-extension-for-pytorch==2.7.0",
    "oneccl-bind-pt==2.7.0",
    "datasets==4.0.*",
    "transformers==4.56.*",
    "evaluate==0.4.*",
    "tqdm>=4.67.1",
    "scikit-learn>=1.7.1",
    "accelerate>=1.10.1",
    "sentencepiece>=0.2.1",
    "matplotlib>=3.10.6",
]

[dependency-groups]
dev = [
    "ruff~=0.4.10",
    "mypy>=1.10.0",
    "pytest>=6.0",
]


[tool.uv.sources]
# Direct URL overrides for CPU-specific Torch wheels
torch = { url = "https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp312-cp312-manylinux_2_28_x86_64.whl" }
torchvision = { url = "https://download.pytorch.org/whl/cpu/torchvision-0.22.0%2Bcpu-cp312-cp312-manylinux_2_28_x86_64.whl" }
# Ensure oneccl pulls from the Intel index defined in tool.uv
oneccl-bind-pt = { index = "intel" }

[[tool.uv.index]]
name = "intel"
url = "https://pytorch-extension.intel.com/release-whl/stable/cpu/us/"
explicit = true # Only use this index for packages explicitly assigned to it

[tool.pytest.ini_options]
minversion = "6.0"
addopts = ["-ra", "--showlocals", "--strict-markers", "--strict-config"]
xfail_strict = true
filterwarnings = ["error"]
testpaths = ["tests"]

[tool.ruff]
extend-select = [
  "B",    # flake8-bugbear
  "B904",
  "I",    # isort
  "PGH",  # pygrep-hooks
  "RUF",  # Ruff-specific
  "UP",   # pyupgrade
]
extend-ignore = [
  "E501",   # Line too long
  "E741",   # Ambiguous variable name
]
target-version = "py311"
line-length = 120

[tool.ruff.lint]
select = ["E4", "E7", "E9", "F"]
fixable = ["ALL"]

[tool.ruff.lint.isort]
lines-after-imports = 2
lines-between-types = 1
known-first-party = ["mlops_inference_workspace"]

[tool.ruff.format]
quote-style = "double"
indent-style = "space"