lint

tonyf · tonyf · commit adaedd00ff9a · 2025-08-11T11:03:52.000-04:00
diff --git a/torch/distributed/fsdp/_fully_shard/_fsdp_param_group.py b/torch/distributed/fsdp/_fully_shard/_fsdp_param_group.py
@@ -454,7 +454,9 @@ def post_backward(self, *unused: Any):
                     fsdp_param.unsharded_accumulated_grad = None
                 elif fsdp_param.unsharded_param.grad is not None:
                     fsdp_params_with_grad.append(fsdp_param)
-                    unsharded_grads.append(fsdp_param.unsharded_grad_data.to(fsdp_param.reduce_dtype))
+                    unsharded_grads.append(
+                        fsdp_param.unsharded_grad_data.to(fsdp_param.reduce_dtype)
+                    )
                     fsdp_param.unsharded_param.grad = None
             if self.reshard_after_backward:
                 self.reshard()
