pyright clean

mosaicml · Aug 23, 2023 · 290d46c · 290d46c
1 parent ab836e9
commit 290d46c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/models/utils/bert_padding.py b/llmfoundry/models/utils/bert_padding.py
@@ -161,6 +161,6 @@ def pad_input(hidden_states: torch.Tensor, indices: torch.Tensor, batch: int,
     """
     output = index_put_first_axis(hidden_states, indices, batch * seqlen)
     return rearrange(
-        output,
+        output, #pyright: ignore[reportGeneralTypeIssues]
         '(b s) ... -> b s ...',  #pyright: ignore[reportGeneralTypeIssues]
         b=batch)