Fix precommit format issues (#3587)

Summary:
Pull Request resolved: #3587

Fix pre-commit format issues:
https://github.com/meta-pytorch/torchrec/actions/runs/19862546785/job/56916260856

Reviewed By: TroyGarden

Differential Revision: D88212265

fbshipit-source-id: 6000f874a2926f50b990f0aef2c6d88061cbf263

Author: Shuangping Liu

torchrec/distributed/batched_embedding_kernel.py

@@ -2682,7 +2682,8 @@ def _reduce_scatter_weights_async(self) -> ReduceScatterResizeAwaitable:
             self._async_event.record(self._async_stream)
 
             def resize_callback() -> None:
-                self._emb_module.weights_dev.untyped_storage().resize_(0)  # pyre-ignore[29]
+                # pyre-ignore[29]
+                self._emb_module.weights_dev.untyped_storage().resize_(0)
                 self._emb_module.weights_dev = self._shard_buf  # pyre-ignore[16]
 
             return ReduceScatterResizeAwaitable(
@@ -3740,7 +3741,8 @@ def _reduce_scatter_weights_async(self) -> ReduceScatterResizeAwaitable:
             self._async_event.record(self._async_stream)
 
             def resize_callback() -> None:
-                self._emb_module.weights_dev.untyped_storage().resize_(0)  # pyre-ignore[29]
+                # pyre-ignore[29]
+                self._emb_module.weights_dev.untyped_storage().resize_(0)
                 self._emb_module.weights_dev = self._shard_buf  # pyre-ignore[16]
 
             return ReduceScatterResizeAwaitable(