diff --git a/mypy.ini b/mypy.ini
index 7c4eb3e..b5896ba 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -20,4 +20,7 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 
 [mypy-docopt.*]
+ignore_missing_imports = True
+
+[mypy-torch.*]
 ignore_missing_imports = True
\ No newline at end of file