fix unrequred type ignore statements

flairNLP · Oct 25, 2024 · a30fd78 · a30fd78
1 parent 4202333
commit a30fd78
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flair/embeddings/document.py b/flair/embeddings/document.py
@@ -371,7 +371,7 @@ def _add_embeddings_internal(self, sentences: list[Sentence]):
             sentence_tensor = self.word_reprojection_map(sentence_tensor)
 
         # push through RNN
-        packed = pack_padded_sequence(sentence_tensor, lengths, enforce_sorted=False, batch_first=True)  # type: ignore[arg-type]
+        packed = pack_padded_sequence(sentence_tensor, lengths, enforce_sorted=False, batch_first=True)
         rnn_out, hidden = self.rnn(packed)
         outputs, output_lengths = pad_packed_sequence(rnn_out, batch_first=True)
 

diff --git a/flair/embeddings/token.py b/flair/embeddings/token.py
@@ -513,7 +513,7 @@ def _add_embeddings_internal(self, sentences: list[Sentence]):
 
             character_embeddings = self.char_embedding(chars).transpose(0, 1)
 
-            packed = torch.nn.utils.rnn.pack_padded_sequence(character_embeddings, chars2_length)  # type: ignore[arg-type]
+            packed = torch.nn.utils.rnn.pack_padded_sequence(character_embeddings, chars2_length)
 
             lstm_out, self.hidden = self.char_rnn(packed)