From 4b598a1b5438384f7f3405c298f9f24d643dfd92 Mon Sep 17 00:00:00 2001 From: richardpaulhudson Date: Fri, 9 Dec 2022 16:52:19 +0100 Subject: [PATCH] Fix mypy issues --- spacy/pipeline/edit_tree_lemmatizer.py | 2 +- spacy/tests/pipeline/test_edit_tree_lemmatizer.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/spacy/pipeline/edit_tree_lemmatizer.py b/spacy/pipeline/edit_tree_lemmatizer.py index 2c43c996b..d8279da31 100644 --- a/spacy/pipeline/edit_tree_lemmatizer.py +++ b/spacy/pipeline/edit_tree_lemmatizer.py @@ -163,7 +163,7 @@ class EditTreeLemmatizer(TrainablePipe): for i, doc_d_tree_scores in enumerate(d_tree_scores): eg_lowercasing_flags = lowercasing_flags[i] eg_d_lowercasing_flags, eg_lowercasing_loss = lowercasing_loss_func( - eg_lowercasing_flags, self.model.ops.asarray2f(lowercasing_truths[i]) + eg_lowercasing_flags, self.model.ops.asarray2i(lowercasing_truths[i]) ) doc_d_scores = self.model.ops.xp.hstack( [ diff --git a/spacy/tests/pipeline/test_edit_tree_lemmatizer.py b/spacy/tests/pipeline/test_edit_tree_lemmatizer.py index eb8127c34..7197f688c 100644 --- a/spacy/tests/pipeline/test_edit_tree_lemmatizer.py +++ b/spacy/tests/pipeline/test_edit_tree_lemmatizer.py @@ -1,4 +1,4 @@ -from typing import cast, Dict +from typing import cast, Dict, Optional import pickle import pytest from hypothesis import given @@ -193,7 +193,7 @@ def test_incomplete_data(lowercasing: bool): train_examples.append(Example.from_dict(nlp.make_doc(t[0]), t[1])) optimizer = nlp.initialize(get_examples=lambda: train_examples) for i in range(50): - losses: Dict[Floats2d, Floats2d] = {} + losses: Optional[Dict[str, float]] = {} nlp.update(train_examples, sgd=optimizer, losses=losses) # test the trained model @@ -222,7 +222,7 @@ def test_overfitting_IO(lowercasing: bool): optimizer = nlp.initialize(get_examples=lambda: train_examples) for i in range(50): - losses: Dict[Floats2d, Floats2d] = {} + losses: Optional[Dict[str, float]] = {} nlp.update(train_examples, sgd=optimizer, losses=losses) test_text = "She likes blue eggs" @@ -411,7 +411,7 @@ def test_lowercasing(lowercasing: bool): train_examples.append(Example.from_dict(nlp.make_doc(t[0]), t[1])) optimizer = nlp.initialize(get_examples=lambda: train_examples) for _ in range(50): - losses: Dict[Floats2d, Floats2d] = {} + losses: Optional[Dict[str, float]] = {} nlp.update(train_examples, sgd=optimizer, losses=losses) # test the trained model