Fix the lint (#15973)
Summary:
Fix the lint error introduced in https://github.com/pytorch/pytorch/pull/15965
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15973
Differential Revision: D13640856
Pulled By: houseroad
fbshipit-source-id: 3f14d9898dcfb0fc469468f63fa1461c88b66b2e