Skip to content

Commit f431b71

Browse files
q10meta-codesync[bot]
authored andcommitted
Fix broken fbcode//deeplearning/fbgemm/fbgemm_gpu/test/sparse:permute_embeddings-library-type-checki (#5808)
Summary: X-link: https://github.com/facebookresearch/FBGEMM/pull/2734 Pull Request resolved: #5808 Reviewed By: henrylhtsang Differential Revision: D106786915 fbshipit-source-id: d49d3ad2ac44afe912039ae80ad22fc4ec486a5f
1 parent 3df231b commit f431b71

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

fbgemm_gpu/test/sparse/permute_embeddings_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,8 @@ def test_permute_embeddings(
6464
) -> None:
6565
index_dtype = torch.int64 if long_index else torch.int32
6666
lengths = torch.randint(low=1, high=L, size=(T, B)).type(index_dtype)
67+
# pyre-fixme[6]: For 1st param expected `list[int] | Size |
68+
# tuple[int, ...]` but got `bool | float | int`.
6769
embeddings = torch.rand(lengths.sum().item()).float()
6870
permute_list = list(range(T))
6971
random.shuffle(permute_list)

0 commit comments

Comments
 (0)