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

small type fixes

Browse Source
This commit is contained in:
svlandeg 2022-05-25 13:50:54 +02:00
parent cea40c9d7b
commit aa2eb2789c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/pipeline/span_predictor.py
View File

@ -96,7 +96,7 @@ class SpanPredictor(TrainablePipe):
self.input_prefix = input_prefix self.input_prefix = input_prefix
self.output_prefix = output_prefix self.output_prefix = output_prefix
self.cfg = {} self.cfg: Dict[str, Any] = {}
def predict(self, docs: Iterable[Doc]) -> List[MentionClusters]: def predict(self, docs: Iterable[Doc]) -> List[MentionClusters]:
# for now pretend there's just one doc # for now pretend there's just one doc
@ -205,7 +205,7 @@ class SpanPredictor(TrainablePipe):
ops = self.model.ops ops = self.model.ops
# NOTE This is doing fake batching, and should always get a list of one example # NOTE This is doing fake batching, and should always get a list of one example
assert len(examples) == 1, "Only fake batching is supported." assert len(list(examples)) == 1, "Only fake batching is supported."
# starts and ends are gold starts and ends (Ints1d) # starts and ends are gold starts and ends (Ints1d)
# span_scores is a Floats3d. What are the axes? mention x token x start/end # span_scores is a Floats3d. What are the axes? mention x token x start/end
for eg in examples: for eg in examples: