gpauloski/kfac-pytorch

Add `torch` as mypy additional dependency

Closed this issue · 0 comments

Describe the Request

The mypy run in pre-commit does not have torch installed so cannot type check anything related to torch. Note that adding torch will result in a lot of mypy errors popping up.

     hooks:
     -   id: mypy
+        additional_dependencies: [torch]

Sample Code

No response