rocm_jax/mypy.ini

[mypy]
show_error_codes=True

[mypy-absl.*]
ignore_missing_imports = True
[mypy-jaxlib.*]
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
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]`
Fixed mypy type errors for numpy 1.20 Revert also previous changes that pinned numpy to 1.19. One of the changes in numpy 1.20 is to add more type annotations. However, this sometimes make mypy give errors. A common example is numpy.take, which with the new type annotation does not appear to mypy as indexable. Another change is that np.int and np.bool are deprecated. One should use np.bool_ or np.int_, or the built-ins bool and int. 2021-01-31 15:34:20 +02:00			`show_error_codes=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`
			`[mypy-jaxlib.*]`
			`ignore_missing_imports = True`
			`[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`