pytest.ini

[pytest]

filterwarnings =
# By default error out on any warnings.
    error
# Jax warning when no gpu/tpu found.
    ignore:No GPU/TPU found, falling back to CPU.*:UserWarning
# We should remove our dependence on jtu.
    ignore:jax.test_util.check_eq is deprecated and will soon be removed.*:FutureWarning
# Tensorflow's fast_tensor_util.pyx cython raises:
# ImportWarning: can't resolve package from __spec__ or __package__, falling back on __name__ and __path__
    ignore:can't resolve package from.*:ImportWarning
# traverse_util.Traversal will be removed soon.
    ignore:`flax.traverse_util.Traversal` will be deprecated.*:DeprecationWarning
# distutils DeprecationWarning
    ignore:distutils Version classes are deprecated.*:DeprecationWarning
# Deprecated numpy for Python 3.9
    ignore:`np.bool.*` is a deprecated alias:DeprecationWarning
    ignore:NumPy will stop allowing conversion of out-of-bound Python integers to integer arrays:DeprecationWarning
# Deprecated sharding symbol
    ignore:jax.experimental.maps.Mesh is deprecated. Use jax.sharding.Mesh.*:DeprecationWarning
# Deprecated legacy checkpoint - just want to keep the tests running for a while
    ignore:Flax Checkpointing will soon be deprecated in favor of Orbax.*:DeprecationWarning
# Some Tensorflow IO error on 3/27/2023
    ignore:file system plugins are not loaded.*:UserWarning
    ignore:unable to load libtensorflow_io_plugins.so.*:UserWarning
# Remove this after next Optax release after 3/27/2023
    ignore:jax.numpy.DeviceArray is deprecated. Use jax.Array.*:DeprecationWarning