[FSDP][Easy] ufmt files (#90548)
```
ufmt format torch/distributed/fsdp
ufmt format test/distributed/fsdp
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/90548
Approved by: https://github.com/mrshenli, https://github.com/rohan-varma