Add mypy and typing_copilot checks to the CI pipeline.

.azure-pipelines/azure-pipelines-external.yml

@@ -127,6 +127,14 @@ jobs:
       python -m black --check arviz examples asv_benchmarks
     displayName: 'black'
 
+  - script: |
+      python -m mypy .
+    displayName: 'mypy'
+
+  - script: |
+      python -m typing_copilot.cli tighten --error-if-can-tighten
+    displayName: 'typing_copilot'
+
   - script: |
       pytest arviz/tests/helpers.py
     displayName: 'precompile models'

mypy.ini

@@ -0,0 +1,285 @@
+
+# Autogenerated by typing_copilot v0.5.4
+[mypy]
+no_implicit_optional = True
+strict_optional = True
+warn_redundant_casts = True
+check_untyped_defs = True
+disallow_untyped_calls = True
+disallow_incomplete_defs = True
+disallow_untyped_defs = True
+disallow_untyped_decorators = True
+ignore_missing_imports = False
+warn_unused_ignores = True
+
+
+# First party per-module rule relaxations
+[mypy-arviz.*]
+disallow_untyped_calls = False
+disallow_untyped_defs = False
+
+[mypy-arviz.data.datasets.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.inference_data.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_cmdstan.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_cmdstanpy.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_dict.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_emcee.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_json.*]
+check_untyped_defs = False
+
+[mypy-arviz.data.io_numpyro.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_pyjags.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_pymc3.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_pyro.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_pystan.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.data.io_tfp.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.labels.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.plots.backends.*]
+check_untyped_defs = False
+
+[mypy-arviz.plots.backends.bokeh.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.plots.backends.matplotlib.forestplot.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.plots.backends.matplotlib.posteriorplot.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.plots.backends.matplotlib.ppcplot.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.plots.densityplot.*]
+check_untyped_defs = False
+
+[mypy-arviz.plots.elpdplot.*]
+check_untyped_defs = False
+
+[mypy-arviz.plots.pairplot.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.plots.separationplot.*]
+check_untyped_defs = False
+
+[mypy-arviz.plots.traceplot.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.rcparams.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.sel_utils.*]
+check_untyped_defs = False
+
+[mypy-arviz.stats.density_utils.*]
+check_untyped_defs = False
+
+[mypy-arviz.stats.diagnostics.*]
+check_untyped_defs = False
+
+[mypy-arviz.stats.stats.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.stats.stats_utils.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_data.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_diagnostics.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_diagnostics_numba.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_plot_utils.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_plots_bokeh.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_plots_matplotlib.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_rcparams.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_stats.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_stats_numba.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_stats_utils.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_utils.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.base_tests.test_utils_numba.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.conftest.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_cmdstan.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_cmdstanpy.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_emcee.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_numpyro.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_pyjags.*]
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_pymc.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_pyro.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.external_tests.test_data_pystan.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.tests.helpers.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-arviz.utils.*]
+check_untyped_defs = False
+disallow_incomplete_defs = False
+
+[mypy-setup.*]
+disallow_incomplete_defs = False
+disallow_untyped_calls = False
+disallow_untyped_defs = False
+
+
+# Third-party module rule relaxations
+[mypy-IPython.*]
+ignore_missing_imports = True
+
+[mypy-aesara.*]
+ignore_missing_imports = True
+
+[mypy-bokeh.*]
+ignore_missing_imports = True
+
+[mypy-cmdstanpy.*]
+ignore_missing_imports = True
+
+[mypy-emcee.*]
+ignore_missing_imports = True
+
+[mypy-jax.*]
+ignore_missing_imports = True
+
+[mypy-matplotlib.*]
+ignore_missing_imports = True
+
+[mypy-mpl_toolkits.axes_grid1.*]
+ignore_missing_imports = True
+
+[mypy-netCDF4.*]
+ignore_missing_imports = True
+
+[mypy-numba.*]
+ignore_missing_imports = True
+
+[mypy-numpy.*]
+ignore_missing_imports = True
+
+[mypy-numpyro.*]
+ignore_missing_imports = True
+
+[mypy-pandas.*]
+ignore_missing_imports = True
+
+[mypy-pyjags.*]
+ignore_missing_imports = True
+
+[mypy-pymc3.*]
+ignore_missing_imports = True
+
+[mypy-pyro.*]
+ignore_missing_imports = True
+
+[mypy-pystan.*]
+ignore_missing_imports = True
+
+[mypy-scipy.*]
+ignore_missing_imports = True
+
+[mypy-setuptools.*]
+ignore_missing_imports = True
+
+[mypy-stan.*]
+ignore_missing_imports = True
+
+[mypy-tensorflow.*]
+ignore_missing_imports = True
+
+[mypy-tensorflow_probability.*]
+ignore_missing_imports = True
+
+[mypy-theano.*]
+ignore_missing_imports = True
+
+[mypy-zarr.*]
+ignore_missing_imports = True

requirements-dev.txt

@@ -5,3 +5,5 @@ pylint
 pytest
 pytest-cov
 black ; python_version >= '3.6'
+typing_copilot ; python_version >= '3.7'
+mypy<0.800