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: if gold_lemma is None:
label = -1 label = -1
eg_lowercasing_truths.append([0]) eg_lowercasing_truths.append([0.0])
else: else:
if self.lowercasing and _should_lowercased( if self.lowercasing and _should_lowercased(
predicted.text, gold_lemma predicted.text, gold_lemma
): ):
eg_lowercasing_truths.append([1]) eg_lowercasing_truths.append([1.0])
text = predicted.lower_ text = predicted.lower_
else: else:
eg_lowercasing_truths.append([0]) eg_lowercasing_truths.append([0.0])
text = predicted.text text = predicted.text
tree_id = self.trees.add(text, gold_lemma) tree_id = self.trees.add(text, gold_lemma)
label = self.tree2label.get(tree_id, 0) 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): for i, doc_d_tree_scores in enumerate(d_tree_scores):
eg_lowercasing_flags = lowercasing_flags[i] eg_lowercasing_flags = lowercasing_flags[i]
eg_d_lowercasing_flags, eg_lowercasing_loss = lowercasing_loss_func( 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( doc_d_scores = self.model.ops.xp.hstack(
[ [