Fix Lint (#24381)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24381
As pointed out in https://github.com/pytorch/pytorch/pull/24299#issuecomment-521471089, the previous PR broke the Lint.
ghstack-source-id: 88339887
Reviewed By: jamesr66a
Differential Revision: D16822443
fbshipit-source-id: 3aed5b9404b0f0fcf453c05b59189974243b0df2
