torch: Fix black linter
Fixes formatting issues when trying to import diff train
Signed-off-by: Eli Uriegas <eliuriegasfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77925
Approved by: https://github.com/mehtanirav, https://github.com/osalpekar
