fix format (#3358)

Summary:
Pull Request resolved: #3358

this is a follow up to D81154653. fix code format

Reviewed By: aliafzal

Differential Revision: D81817621

fbshipit-source-id: aea266b14d29ae97a3bbb36dbced4d5ccc29ab22

Author: jiayulu

torchrec/distributed/utils.py

@@ -696,7 +696,7 @@ def quantize_embedding_modules(
         sharded_embs = _group_sharded_modules(module)
         sharded_embs.sort(key=weights_bytes_in_emb_kernel)
         logger.info(
-            f"convert embedding modules to converted_dtype={converted_dtype.value} quantization"
+            f"[TorchRec] Converting embedding modules to converted_dtype={converted_dtype.value} quantization"
         )
         converted_sparse_dtype = data_type_to_sparse_type(converted_dtype)
 
@@ -727,7 +727,9 @@ def recreate_embedding_modules(
         sharded_embs.sort(key=weights_bytes_in_emb_kernel)
 
         for emb_kernel in sharded_embs:
-            converted_sparse_dtype = self._emb_kernel_to_sparse_dtype[emb_kernel]  # pyre-ignore [6]: Incompatible parameter type
+            converted_sparse_dtype = self._emb_kernel_to_sparse_dtype[
+                emb_kernel  # pyre-ignore [6]: Incompatible parameter type
+            ]
 
             emb_kernel.weights_dev = _convert_weights(  # pyre-ignore [16]
                 emb_kernel.weights_dev,  # pyre-ignore [6]