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

Fix Pipe base class

Browse Source
This commit is contained in:
Matthew Honnibal 2019-08-01 17:29:01 +02:00
parent 8718ca8b1f
commit 4632c597e7
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/pipeline/pipes.pyx
View File

@ -66,8 +66,12 @@ class Pipe(object):
and `set_annotations()` methods. and `set_annotations()` methods.
""" """
self.require_model() self.require_model()
scores, tensors = self.predict([doc]) predictions = self.predict([doc])
self.set_annotations([doc], scores, tensors=tensors) if isinstance(predictions, tuple) and len(tuple) == 2:
scores, tensors = predictions
self.set_annotations([doc], scores, tensor=tensors)
else:
self.set_annotations([doc], predictions)
return doc return doc
def require_model(self): def require_model(self):