-
Notifications
You must be signed in to change notification settings - Fork 2.9k
/
Copy pathpyproject.toml
139 lines (129 loc) · 3.68 KB
/
pyproject.toml
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"
[tool.mypy]
show_error_codes = true
disable_error_code = "attr-defined, name-defined, annotation-unchecked"
no_implicit_optional = true
warn_redundant_casts = true
allow_redefinition = true
[[tool.mypy.overrides]]
module = [
"IPython.*",
"absl.*",
"colorama.*",
"etils.*",
"filelock.*",
"flatbuffers.*",
"flax.*",
"google.colab.*",
"hypothesis.*",
"jax.experimental.jax2tf.tests.back_compat_testdata",
"jax.experimental.jax2tf.tests.flax_models",
"jax_cuda12_plugin.*",
"jaxlib.*",
"jaxlib.mlir.*",
"jraph.*",
"libtpu.*",
"matplotlib.*",
"nvidia.*",
"numpy.*",
"opt_einsum.*",
"optax.*",
"pygments.*",
"pytest.*",
"rich.*",
"scipy.*",
"setuptools.*",
"tensorboard_plugin_profile.convert.*",
"tensorflow.*",
"tensorflow.io.*",
"tensorflowjs.*",
"tensorstore.*",
"web_pdb.*",
"zstandard.*",
"kubernetes.*"
]
ignore_missing_imports = true
[tool.pytest.ini_options]
markers = [
"multiaccelerator: indicates that a test can make use of and possibly requires multiple accelerators",
"SlurmMultiNodeGpuTest: mark a test for Slurm multinode GPU nightly CI"
]
filterwarnings = [
"error",
# TODO(jakevdp): remove when array_api_tests stabilize
"default:.*not machine-readable.*:UserWarning",
"default:Special cases found for .* but none were parsed.*:UserWarning",
# jax.profiler imports tensorflow.python.profiler.trace internally, which
# will fail with python 3.12 and some versions of protobuf because the
# "error" entry above promotes DeprecationWarnings into errors. See also:
# https://github.com/protocolbuffers/protobuf/issues/12186#issuecomment-1745679358
"ignore:Type google\\._upb\\._message\\.(Scalar|Message)MapContainer uses PyType_Spec with a metaclass that has custom tp_new\\. This is deprecated and will no longer be allowed in Python 3\\.14\\.:DeprecationWarning",
# NOTE: this is probably not where you want to add code to suppress a
# warning. Only pytest tests look at this list, whereas Bazel tests also
# check for warnings and do not check this list. Most likely, you should
# add a @jtu.ignore_warning decorator to your test instead.
]
doctest_optionflags = [
"NUMBER",
"NORMALIZE_WHITESPACE"
]
addopts = "--doctest-glob='*.rst' --ignore='examples/ffi'"
[tool.ruff]
preview = true
exclude = [
".git",
"build",
"__pycache__",
]
line-length = 88
indent-width = 2
target-version = "py310"
[tool.ruff.lint]
ignore = [
# Unnecessary collection call
"C408",
# Unnecessary map usage
"C417",
# Unnecessary dict comprehension for iterable
"C420",
# Object names too complex
"C901",
# Local variable is assigned to but never used
"F841",
# Raise with from clause inside except block
"B904",
# Zip without explicit strict parameter
"B905",
]
select = [
"B9",
"C",
"F",
"W",
"YTT",
"ASYNC",
"E101",
"E112",
"E113",
"E115",
"E117",
"E225",
"E227",
"E228",
]
[tool.ruff.lint.mccabe]
max-complexity = 18
[tool.ruff.lint.per-file-ignores]
# F811: Redefinition of unused name.
# F821: Undefined name.
"docs/autodidax.py" = ["F811"]
"docs/pallas/tpu/matmul.ipynb" = ["F811"]
"docs/pallas/tpu/distributed.ipynb" = ["F811"]
"docs/pallas/quickstart.ipynb" = ["F811"]
"docs/notebooks/autodiff_cookbook.ipynb" = ["F811", "F821"]
"docs/notebooks/autodiff_remat.ipynb" = ["F811", "F821"]
"docs/notebooks/Custom_derivative_rules_for_Python_code.ipynb" = ["F811"]
"docs/jep/9407-type-promotion.ipynb" = ["F811"]
"docs/autodidax.ipynb" = ["F811"]