Fix lint checker breakage caused by #25111 (#25122)
Summary:
fix lint by flake8
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25122
Differential Revision: D16995103
Pulled By: zrphercule
fbshipit-source-id: 810be4d8073cae73d4b0f6d82b410fd235a73bbb