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

Fix mypy issues

Browse Source
This commit is contained in:
richardpaulhudson 2022-12-09 17:03:48 +01:00
parent 4b598a1b54
commit 449559cc2d
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
spacy/pipeline/edit_tree_lemmatizer.py
View File

@ -137,15 +137,15 @@ class EditTreeLemmatizer(TrainablePipe):
):
if gold_lemma is None:
label = -1
eg_lowercasing_truths.append([0])
eg_lowercasing_truths.append([0.0])
else:
if self.lowercasing and _should_lowercased(
predicted.text, gold_lemma
):
eg_lowercasing_truths.append([1])
eg_lowercasing_truths.append([1.0])
text = predicted.lower_
else:
eg_lowercasing_truths.append([0])
eg_lowercasing_truths.append([0.0])
text = predicted.text
tree_id = self.trees.add(text, gold_lemma)
label = self.tree2label.get(tree_id, 0)
@ -163,7 +163,8 @@ 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.asarray2i(lowercasing_truths[i])
eg_lowercasing_flags,
self.model.ops.asarray2f(lowercasing_truths[i]), # type: ignore[arg-type]
)
doc_d_scores = self.model.ops.xp.hstack(
[