einf/pyproject.toml at develop · isty2e/einf · 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
[build-system]
requires = ["hatch-vcs>=0.4", "hatchling>=1.24"]
build-backend = "hatchling.build"


[project]
name = "einf"
dynamic = ["version"]
description = "Tensor DSL with axis signatures and first-class TensorOp transforms."
requires-python = ">=3.10"
dependencies = [
  "array-api-compat>=1.13",
  "opt_einsum>=3.4",
  "typing_extensions>=4.8; python_version < '3.11'",
]


[project.optional-dependencies]
dev = [
  "array-api-strict>=2.5",
  "basedpyright>=1.31",
  "build>=1.2",
  "einx",
  "einops",
  "mypy>=1.19",
  "numpy>=1.26",
  "pytest>=8.0",
  "ruff==0.16.0",
  "threadpoolctl>=3.5",
  "ty>=0.0.64",
]
analysis = [
  "libcst>=1.0",
]
lsp = [
  "lsprotocol>=2023.0.0",
  "pygls>=1.3",
]
docs = [
  "mkdocs>=1.6",
  "mkdocs-material>=9.5",
  "mkdocstrings[python]>=0.24",
]

[project.scripts]
einf-validate = "einf.analysis.validator.cli:main"
einf-lsp = "einf.analysis.lsp.cli:main"


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

[tool.hatch.build.targets.wheel.force-include]
"src/einf/operations/api.pyi" = "einf/operations/api.pyi"
"src/einf/py.typed" = "einf/py.typed"

[tool.hatch.build.targets.sdist]
include = [
  "src/einf/**/*.py",
  "src/einf/**/*.pyi",
  "src/einf/py.typed",
  "tests/**/*.py",
  "pyproject.toml",
]

[tool.hatch.version]
source = "vcs"
tag-pattern = '^v(?P<version>(?:0|[1-9][0-9]*)\.(?:0|[1-9][0-9]*)\.(?:0|[1-9][0-9]*))$'

[tool.hatch.version.raw-options]
git_describe_command = "git describe --dirty --tags --long --match v[0-9]*.[0-9]*.[0-9]*"

[tool.hatch.build.hooks.vcs]
version-file = "src/einf/_version.py"


[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = "-q"

[tool.basedpyright]
include = ["src", "tests"]
exclude = [
  # Checked by the optional LSP transport job.
  "src/einf/analysis/lsp/cli.py",
  "src/einf/analysis/lsp/hover.py",
  "src/einf/analysis/lsp/inlay_hints.py",
  "src/einf/analysis/lsp/position_codec.py",
  "src/einf/analysis/lsp/server.py",
  "tests/analysis/test_analysis_lsp_position_encoding.py",
  "tests/analysis/test_analysis_lsp_server.py",
  "tests/analysis/test_conformance_analysis_lsp.py",
  # Checked by the optional PyTorch integration job.
  "tests/integration/test_array_api_contract_einop_execution.py",
  "tests/integration/test_array_api_rearrange_execution.py",
  "tests/integration/test_array_api_repeat_reduce_execution.py",
  "tests/integration/test_array_api_view_execution.py",
]
typeCheckingMode = "standard"