diff --git a/Makefile b/Makefile index b531c2a515..04c6112d47 100644 --- a/Makefile +++ b/Makefile @@ -21,7 +21,7 @@ lint: FORCE ruff check . black --check *.py pyro examples tests scripts profiler python scripts/update_headers.py --check - mypy --install-types --non-interactive pyro scripts + mypy --install-types --non-interactive pyro scripts tests license: FORCE python scripts/update_headers.py diff --git a/setup.cfg b/setup.cfg index ff6d3042d8..8e36d8aff2 100644 --- a/setup.cfg +++ b/setup.cfg @@ -82,3 +82,43 @@ warn_unused_ignores = True [mypy-pyro.util.*] ignore_errors = True warn_unused_ignores = True + +[mypy-tests.test_primitives] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.test_generic] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.poutine.*] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.ops.*] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.optim.*] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.perf.*] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.nn.*] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.infer.*] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.distributions.*] +ignore_errors = True +warn_unused_ignores = True + +[mypy-tests.contrib.*] +ignore_errors = True +warn_unused_ignores = True