torch: Fix flake8 errors from leftover import (#56614)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56614
Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
Test Plan: Imported from OSS
Reviewed By: heitorschueroff
Differential Revision: D27917831
Pulled By: seemethere
fbshipit-source-id: 90a3213080cc2c8da2bc63c8971e14f7823390a9