diff --git a/mypy.ini b/mypy.ini index ec5b4fd3f..4083e1379 100644 --- a/mypy.ini +++ b/mypy.ini @@ -1,3 +1,7 @@ [mypy] ignore_missing_imports = true exclude = output + +[mypy-torch.*] +follow_imports = skip +follow_imports_for_stubs = True