explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-24 14:04:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix mypy issues

Browse Source
This commit is contained in:
richardpaulhudson 2022-12-09 16:52:19 +01:00
parent 7cd9a99854
commit 4b598a1b54
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/edit_tree_lemmatizer.py
View File

@ -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(
[

8
spacy/tests/pipeline/test_edit_tree_lemmatizer.py
View File

@ -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