fix rust & mypy errors

flairNLP · Sep 11, 2023 · 3ddc564 · 3ddc564
1 parent f1a4d96
commit 3ddc564
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flair/models/multitask_model.py b/flair/models/multitask_model.py
@@ -116,7 +116,7 @@ def split_batch_to_task_ids(sentences: Union[List[Sentence], Sentence], all_task
                     batch_to_task_mapping[multitask_id.value] = [sentence_id]
         return batch_to_task_mapping
 
-    def evaluate(
+    def evaluate(  # type: ignore[override]
         self,
         data_points,
         gold_label_type: str,

diff --git a/flair/trainers/trainer.py b/flair/trainers/trainer.py
@@ -461,7 +461,7 @@ def train_custom(
             if inspect.isclass(sampler):
                 sampler = sampler()
             # set dataset to sample from
-            sampler.set_dataset(train_data)
+            sampler.set_dataset(train_data)  # type: ignore[union-attr]
             shuffle = False
 
         # this field stores the names of all dynamic embeddings in the model (determined after first forward pass)