mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
102 lines
2.8 KiB
TOML
102 lines
2.8 KiB
TOML
[build-system]
|
|
requires = ["setuptools", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[tool.mypy]
|
|
show_error_codes = true
|
|
disable_error_code = "attr-defined"
|
|
no_implicit_optional = true
|
|
|
|
[[tool.mypy.overrides]]
|
|
module = [
|
|
"absl.*",
|
|
"colorama.*",
|
|
"IPython.*",
|
|
"numpy.*",
|
|
"opt_einsum.*",
|
|
"scipy.*",
|
|
"libtpu.*",
|
|
"jaxlib.mlir.*",
|
|
"iree.*",
|
|
"rich.*",
|
|
"optax.*",
|
|
"flax.*",
|
|
"tensorflow.*",
|
|
"tensorflowjs.*",
|
|
"tensorflow.io.*",
|
|
"tensorstore.*",
|
|
"web_pdb.*",
|
|
"etils.*",
|
|
"google.colab.*",
|
|
"pygments.*",
|
|
"jraph.*",
|
|
"matplotlib.*",
|
|
"tensorboard_plugin_profile.convert.*",
|
|
"jaxlib.*",
|
|
"pytest.*",
|
|
"zstandard.*",
|
|
"jax.experimental.jax2tf.tests.flax_models",
|
|
"jax.experimental.jax2tf.tests.back_compat_testdata"
|
|
]
|
|
ignore_missing_imports = true
|
|
|
|
[[tool.mypy.overrides]]
|
|
module = [
|
|
"jax.interpreters.autospmd",
|
|
"jax.lax.lax_parallel",
|
|
"jax.experimental.jax2tf.tests.primitive_harness"
|
|
]
|
|
ignore_errors = 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",
|
|
"ignore:No GPU/TPU found, falling back to CPU.:UserWarning",
|
|
"ignore:xmap is an experimental feature and probably has bugs!",
|
|
"ignore:the imp module is deprecated in favour of importlib.*:DeprecationWarning",
|
|
"ignore:can't resolve package from __spec__ or __package__:ImportWarning",
|
|
"ignore:Using or importing the ABCs.*:DeprecationWarning",
|
|
"ignore:numpy.ufunc size changed",
|
|
"ignore:.*experimental feature",
|
|
"ignore:The distutils.* is deprecated.*:DeprecationWarning",
|
|
"default:Error reading persistent compilation cache entry for 'jit__lambda_'",
|
|
"default:Error writing persistent compilation cache entry for 'jit__lambda_'",
|
|
"ignore:backend and device argument on jit is deprecated.*:DeprecationWarning",
|
|
# TODO(skyewm): remove when jaxlib >= 0.4.12 is released (needs
|
|
# https://github.com/openxla/xla/commit/fb9dc3db0999bf14c78d95cb7c3aa6815221ddc7)
|
|
"ignore:ml_dtypes.float8_e4m3b11 is deprecated.",
|
|
]
|
|
doctest_optionflags = [
|
|
"NUMBER",
|
|
"NORMALIZE_WHITESPACE"
|
|
]
|
|
addopts = "--doctest-glob='*.rst'"
|
|
|
|
[tool.pylint.master]
|
|
extension-pkg-whitelist = "numpy"
|
|
|
|
[tool.pylint."messages control"]
|
|
disable = [
|
|
"missing-docstring",
|
|
"too-many-locals",
|
|
"invalid-name",
|
|
"redefined-outer-name",
|
|
"redefined-builtin",
|
|
"protected-name",
|
|
"no-else-return",
|
|
"fixme",
|
|
"protected-access",
|
|
"too-many-arguments",
|
|
"blacklisted-name",
|
|
"too-few-public-methods",
|
|
"unnecessary-lambda"
|
|
]
|
|
enable = "c-extension-no-member"
|
|
|
|
[tool.pylint.format]
|
|
indent-string=" "
|