spu/pyproject.toml at main · tyrone-yu/spu · 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
104
105
106
107
108
109
110
111
[build-system]
requires = ["pdm-backend", "packaging"]
build-backend = "pdm.backend"

[[tool.uv.index]]
name = "pypi"
url = "https://pypi.org/simple/"
publish-url = "https://upload.pypi.org/legacy/"
explicit = true

[[tool.uv.index]]
name = "aliyun"
url = "https://mirrors.aliyun.com/pypi/simple/"
default = true

[tool.uv]
default-groups = ["dev"]

[project]
name = "spu"
description = "spu"
readme = "README.md"
license = "Apache-2.0"
authors = [
    { name = "SecretFlow Team", email = "secretflow-contact@service.alipay.com" },
]
classifiers = [
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]
requires-python = ">=3.11,<3.13"
dynamic = ["version"]

dependencies = [
    "numpy>=2.0.0",        # Updated for JAX 0.8.0 compatibility. Note: NumPy 2.0 introduces several breaking changes that may affect downstream users. Please review your code for compatibility with NumPy 2.0 before upgrading.
    "cloudpickle>=2.0.0",
    "multiprocess>=0.70.12.2",
    "cachetools>=5.0.0",
    "jax[cpu]==0.8.0",
    "termcolor>=2.0.0",
]

[dependency-groups]
dev = [
    "pytest",
    "build",
    "protobuf==4.25.8",
    "grpcio>=1.60.0",
    "pandas>=1.4.2",
    "flax==0.10.7",
    "scikit-learn<1.8.0",
    "tensorflow>=2.12.0", # used in spu/utils/frontend.py
]
examples = [
    "dm-haiku",
    "plotnine",
    "jraph",
    "optax",
    "torch==2.3.0",
    "torch-xla==2.3.0; sys_platform == \"linux\" and platform_machine == \"x86_64\"",
    "torchvision",
    "tensorflow-datasets",
    "keras",
    # https://github.com/pypa/setuptools/issues/4487
    "setuptools<71.0.0",
    "transformers",
    "datasets",
    "async-timeout",
]

[project.urls]
Homepage = "https://github.com/secretflow/spu"
Repository = "https://github.com/secretflow/spu"

[tool.pdm.version]
source = "call"
getter = "pdm_build:get_version"

[tool.pdm.build]
includes = ["spu/**/*.py", "spu/**/*.pyi"]
excludes = ["spu/tests/**"]

[tool.pdm.scripts]
build_libs = { call = "pdm_build:build_libs" }

[tool.black]
skip-string-normalization = true
line-length = 88
target-version = ['py311', 'py312']
include = '\.pyi?$'
extend-exclude = '''.*_pb2.py|.*_pb2_grpc.py'''

[tool.isort]
profile = "black"
skip_glob = ["external/*", ".cache/*", "bazel-*/*"]

[tool.rstcheck]
report_level = "ERROR"
ignore_directives = ["include", "mermaid", "autoclass", "autofunction"]
ignore_roles = ["spu_code_host"]
ignore_languages = ["cpp"]

[tool.pyright]
include = ["spu", "examples"]

reportMissingImports = true
reportMissingTypeStubs = false

pythonVersion = "3.11"
pythonPlatform = "Linux"