nkipy/pyproject.toml at main · aws-neuron/nkipy · 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
# NKIPy Monorepo

[project]
name = "nkipy-workspace"
version = "0.1.0"
description = "NKIPy Monorepo - A lightweight Pythonic kernel environment for AWS Neuron"
requires-python = ">=3.10"
dependencies = ["nkipy"]

[tool.uv.workspace]
members = ["nkipy", "spike"]

[[tool.uv.index]]
name = "neuron"
url = "https://pip.repos.neuron.amazonaws.com"
explicit = true

[tool.uv.sources]
spike = { workspace = true }
nkipy = { workspace = true }

[dependency-groups]
dev = ["spike"]
test = [
    "pytest>=7.0",
    "pytest-cov>=4.0",
    "pytest-order>=1.0",
    "pytest-xdist>=3.0",
    "ruff",
    "mypy",
]
examples = [
    "torch",
    "ipython",
    "transformers",
    "argparse",
    "safetensors",
    "accelerate",
    "boto3>=1.34",
]
docs = [
    "sphinx",
    "sphinx-book-theme",
    "myst-parser",
    "myst-nb",
    "sphinx-autodoc-typehints",
]

[tool.ruff]
line-length = 88
exclude = ["build", "third-party", "spike/src/spike/_spike.pyi"]

[tool.ruff.lint]
select = ["E", "F", "I"]

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

[tool.pytest.ini_options]
testpaths = ["tests", "spike/tests"]

[tool.mypy]
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = false

[[tool.mypy.overrides]]
module = "neuronxcc.*"
ignore_missing_imports = true

[tool.coverage.run]
source = ["nkipy/src/nkipy", "spike/src/spike"]
branch = true

[tool.coverage.report]
show_missing = true

[tool.coverage.html]
directory = "htmlcov"