pytorch

4d3930fe - Fix mypy lint failure

Fix mypy lint failure Signed-off-by: Edward Z. Yang <ezyangfb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/77652 Approved by: https://github.com/SherlockNoMad