rocm_jax/mypy.ini

[mypy]
show_error_codes = True
disable_error_code = attr-defined
no_implicit_optional = True

[mypy-absl.*]
ignore_missing_imports = True
[mypy-colorama.*]
ignore_missing_imports = True
[mypy-numpy.*]
ignore_missing_imports = True
[mypy-opt_einsum.*]
ignore_missing_imports = True
[mypy-scipy.*]
ignore_missing_imports = True
[mypy-jax.interpreters.autospmd]
ignore_errors = True
[mypy-jax.lax.lax_parallel]
ignore_errors = True
[mypy-jax.experimental.jax2tf.tests.primitive_harness]
ignore_errors = True
[mypy-libtpu.*]
ignore_missing_imports = True
[mypy-jaxlib.mlir.*]
ignore_missing_imports = True
[mypy-iree.*]
ignore_missing_imports = True
Add mypy type checking (#2430) * Add type annotations to make mypy pass. * Add mypy to .travis.yml. 2020-03-18 17:06:05 -04:00			`[mypy]`
unify configuration state handling 2021-03-19 13:49:38 -07:00			`show_error_codes = True`
			`disable_error_code = attr-defined`
Use --no-implicit-optional for type checking This makes type checking PEP 484 compliant (as of 2018). mypy will change its defaults soon. See: https://github.com/python/mypy/issues/9091 https://github.com/python/mypy/pull/13401 2022-08-12 19:07:21 -07:00			`no_implicit_optional = True`
Add mypy type checking (#2430) * Add type annotations to make mypy pass. * Add mypy to .travis.yml. 2020-03-18 17:06:05 -04:00
			`[mypy-absl.*]`
			`ignore_missing_imports = True`
Adds a Wadler-Lindig pretty printer. Changes jaxpr printing to use it. 2021-09-24 22:08:42 -04:00			`[mypy-colorama.*]`
			`ignore_missing_imports = True`
Add mypy type checking (#2430) * Add type annotations to make mypy pass. * Add mypy to .travis.yml. 2020-03-18 17:06:05 -04:00			`[mypy-numpy.*]`
			`ignore_missing_imports = True`
			`[mypy-opt_einsum.*]`
			`ignore_missing_imports = True`
			`[mypy-scipy.*]`
			`ignore_missing_imports = True`
add type annotations to core.py tracing machinery also add .copy() method to core.trace_state global trace state 2020-03-28 14:55:58 -07:00			`[mypy-jax.interpreters.autospmd]`
			`ignore_errors = True`
omnistaging, under a flag and disabled by default (#3370) This change, when enabled, stages out all primitive calls in the dynamic scope of a jitted, pmapped, or control flow function, rather than only staging out based on data dependence. One improvement is that jitted functions can consume less memory, by avoiding instantiating large constants at trace time, and cause less memory fragmentation as well. It also simplifies several internals. See https://github.com/google/jax/pull/3370 fo more information. 2020-07-30 12:59:36 -07:00			`[mypy-jax.lax.lax_parallel]`
			`ignore_errors = True`
Major refactoring of the jax2tf test harnesses. See the PR description for details. 2020-12-24 09:40:40 +02:00			`[mypy-jax.experimental.jax2tf.tests.primitive_harness]`
			`ignore_errors = True`
Fix type errors with current mypy and NumPy. Enable type stubs for jaxlib. Fix a nondeterminism problem in jax2tf tests. 2021-06-24 10:51:06 -04:00			`[mypy-libtpu.*]`
			`ignore_missing_imports = True`
Adds an initial prototype of an alternate JAX compilation path that emits the MLIR MHLO/CHLO dialects instead of classic XLA HLO. This lowering is missing a number of features, but it is complete enough that many tests pass, and that I would like to start checking it in. PiperOrigin-RevId: 409134016 2021-11-11 06:36:31 -08:00			`[mypy-jaxlib.mlir.*]`
			`ignore_missing_imports = True`
Add a prototype IREE backend for JAX. This is to support experimentation with the combination of JAX/IREE. Many things do not work yet. PiperOrigin-RevId: 409980064 2021-11-15 07:56:34 -08:00			`[mypy-iree.*]`
			`ignore_missing_imports = True`