rocm_jax/pytest.ini

[pytest]
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
doctest_optionflags = NUMBER NORMALIZE_WHITESPACE
addopts = --doctest-glob="*.rst"
Make pytest run over JAX tests warning clean, and error on warnings. (#2674) * Make pytest run over JAX tests warning clean, and error on warnings. Remove global warning suppression in travis.yml. Instead add a pytest.ini that converts warnings to errors, with the exception of a whitelist. Either fix or locally suppress warnings in tests. Also fix crashes on Mac related to a preexisting linear algebra bug. * Fix some type errors in the FFT transpose rules revealed by the convert_element_type transpose rule change. 2020-04-12 15:35:35 -04:00			`[pytest]`
			`filterwarnings =`
			`error`
			`ignore:No GPU/TPU found, falling back to CPU.:UserWarning`
Refactored host_callback to use the C++ runtime. (#3644) * Refactored host_callback to use the C++ runtime. * The new runtime makes it unnecessary to start the outfeed_receiver in the user's code * We don't need msgpack anymore * There is an interaction between host_callback and using lax.outfeed. I am trying to solve this by (a) making host_callback_test stop the outfeed receiver on finish and infeed_test on start, and (b) telling pytest-xdist to run all the tests from one file into a single worker. 2020-07-04 18:12:58 +03:00			`ignore:outfeed_receiver is unnecessary and deprecated:DeprecationWarning`
Run doctest on all source files except jax2tf 2021-04-02 16:43:10 -07:00			`# xmap`
			`ignore:xmap is an experimental feature and probably has bugs!`
Future warning about lists and tuples (#3369) 2020-06-24 11:54:06 -03:00			`# The rest are for experimental/jax_to_tf`
Initial import of jax2tf into JAX core (#3202) * Initial import of jax2tf into JAX core Renamed jax2tf.convert to jax_to_tf. Added Travis test support. Added OSS build configuration. * Added support for squeeze 2020-05-29 09:56:32 +03:00			`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`
Cleanup the API, and more documentation 2021-04-06 11:43:06 +03:00			`# jax2tf tests due to mix of JAX and TF`
			`ignore:numpy.ufunc size changed`
Use pytest's filterwarnings feature instead of filtering each test case. We often forget to put the per-test-case decorators, resulting in test failures in cases not covered by github CI (e.g. Cloud TPU tests). This change filters the "experimental feature" warnings by default. 2021-04-21 21:31:57 +00:00			`ignore:.*experimental feature`
Add DeprecationWarnings to jax.ops.index_... operators. Remove uses of index_... in Common Gotchas notebook. 2021-10-05 16:12:52 -04:00			`ignore:index.is deprecated.:DeprecationWarning`
[JAX] move example libraries from `jax.experimental` into `jax.example_libraries` The `jax.experimental.stax` and `jax.experimental.optimizers` modules are standalone examples libraries. By contrast, the remaining modules in `jax.experimental` are experimental features of the JAX core system. This change moves the two example libraries, and the README that describes them, to `jax.example_libraries` to reflect this distinction. PiperOrigin-RevId: 404405186 2021-10-19 17:30:16 -07:00			`ignore:jax.experimental.* is deprecated, import jax.example_libraries.* instead:FutureWarning`
Test code in docs and api.py docstrings (#2994) Also remove jaxpr doc tests from api_test.py. 2020-05-16 14:19:24 +01:00			`doctest_optionflags = NUMBER NORMALIZE_WHITESPACE`
Make infeed_test and host_callback_test independent. (#3676) * Make infeed_test and host_callback_test independent. * the infeed_test will stop the outfeed receiver * Remove the use of --dist=loadfile. * Prevent logging on exit 2020-07-07 11:03:30 +03:00			`addopts = --doctest-glob="*.rst"`