From 9f519fb5914bd019f15d59bdc41bfce25d67106d Mon Sep 17 00:00:00 2001 From: Pyre Bot Jr Date: Mon, 29 Jan 2024 10:36:18 -0800 Subject: [PATCH] suppress errors in `torchrec` Differential Revision: D53166035 --- torchrec/distributed/quant_embedding.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchrec/distributed/quant_embedding.py b/torchrec/distributed/quant_embedding.py index 1c02c54b1..311336594 100644 --- 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))