pytorch

398a5f4d - Clean up mypy (#102555)

Clean up mypy (#102555) Pull Request resolved: https://github.com/pytorch/pytorch/pull/102555 Approved by: https://github.com/Skylion007