[Easy] Apply black to format _spmd files (#97534)
No real changes. Format code to prepare for the PR on top.
Differential Revision: [D44376380](https://our.internmc.facebook.com/intern/diff/D44376380)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97534
Approved by: https://github.com/wanchaol
