pytorch

e4fba752 - fix type annotation

fix type annotation Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/26930 Test Plan: Imported from OSS Differential Revision: D17614745 Pulled By: vincentqb fbshipit-source-id: 1c29543f74d9cf307e9665aa890b4830b886fe63