suppress errors in torchrec

Differential Revision: D53166035
henrylhtsang · Jan 29, 2024 · 9f519fb · 9f519fb
1 parent 4376dff
commit 9f519fb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchrec/distributed/quant_embedding.py b/torchrec/distributed/quant_embedding.py
@@ -189,6 +189,8 @@ def _construct_jagged_tensors_cw(
     for i in range(len(features)):
         embedding = embeddings[i]
         feature = features[i]
+        # pyre-fixme[6]: For 1st argument expected `List[Tensor]` but got
+        #  `Tuple[Tensor, ...]`.
         lengths_lists.append(torch.unbind(feature.lengths().view(-1, stride), dim=0))
         embeddings_lists.append(
             list(torch.split(embedding, feature.length_per_key(), dim=0))