https://github.com/google/jax
Raw File
Tip revision: 2cbceb4a616ac6ca3bfdb623d2583fc6120e2340 authored by jax authors on 21 October 2022, 22:20:53 UTC
[Flaxformer] Additional asserts to help find confusing misconfigurations of models (added in zT5 testing).
Tip revision: 2cbceb4
pytest.ini
[pytest]
markers =
    SlurmMultiNodeGpuTest: mark a test for Slurm multinode GPU nightly CI
filterwarnings =
    error
    ignore:No GPU/TPU found, falling back to CPU.:UserWarning
    ignore:outfeed_receiver is unnecessary and deprecated:DeprecationWarning
    # xmap
    ignore:xmap is an experimental feature and probably has bugs!
    # The rest are for experimental/jax_to_tf
    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
    # jax2tf tests due to mix of JAX and TF
    ignore:numpy.ufunc size changed
    ignore:.*experimental feature
    ignore:index.*is deprecated.*:DeprecationWarning
    ignore:jax.experimental.* is deprecated, import jax.example_libraries.* instead:FutureWarning
    # numpy uses distutils which is deprecated
    ignore:The distutils.* is deprecated.*:DeprecationWarning
    ignore:`sharded_jit` is deprecated. Please use `pjit` instead.*:DeprecationWarning
doctest_optionflags = NUMBER NORMALIZE_WHITESPACE
addopts = --doctest-glob="*.rst"
back to top