Fix pre-commit

Summary: Fix github ci error caused by D53139358 Differential Revision: D53247357
henrylhtsang · Jan 31, 2024 · 46fc366 · 46fc366
1 parent 5fc2dfa
commit 46fc366
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchrec/sparse/jagged_tensor.py b/torchrec/sparse/jagged_tensor.py
@@ -579,7 +579,9 @@ def _jt_flatten_with_keys(
 ) -> Tuple[List[Tuple[KeyEntry, Optional[torch.Tensor]]], None]:
     values, context = _jt_flatten(t)
     # pyre can't tell that GetAttrKey implements the KeyEntry protocol
-    return [(GetAttrKey(k), v) for k, v in zip(JaggedTensor._fields, values)], context  # pyre-ignore[7]
+    return [  # pyre-ignore[7]
+        (GetAttrKey(k), v) for k, v in zip(JaggedTensor._fields, values)
+    ], context
 
 
 def _jt_unflatten(values: List[Optional[torch.Tensor]], context: None) -> JaggedTensor: