[Easy] Lint wrap.py file (#70341)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70341
Per title
ghstack-source-id: 146181936
Test Plan: CI
Reviewed By: zhaojuanmao
Differential Revision: D33290099
fbshipit-source-id: e4415a42086d9b1b78b0b5f42d4b02f275131dfa