fix lint (#58203)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58203
Test Plan: Imported from OSS
Reviewed By: mruberry
Differential Revision: D28401974
Pulled By: suo
fbshipit-source-id: cc244e0fc81c5f699ff4bd30754a3f6467f232c4
