Fix component check in self.factories (see #1911)

This commit is contained in:
ines 2018-01-30 16:09:37 +01:00
parent 8901814248
commit ce10d320c4

View File

@ -239,7 +239,7 @@ class Language(object):
if not hasattr(component, '__call__'):
msg = ("Not a valid pipeline component. Expected callable, but "
"got {}. ".format(repr(component)))
if component in self.factories.keys():
if component in self.factories:
msg += ("If you meant to add a built-in component, use "
"create_pipe: nlp.add_pipe(nlp.create_pipe('{}'))"
.format(component))