Fix allow_overlap default for spancat scoring (#8970)

* Remove irrelevant default options
This commit is contained in:
Adriane Boyd 2021-08-18 09:56:56 +02:00 committed by GitHub
parent b18cb1cd2a
commit 6722dc3dc5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -386,9 +386,7 @@ class SpanCategorizer(TrainablePipe):
kwargs = dict(kwargs) kwargs = dict(kwargs)
attr_prefix = "spans_" attr_prefix = "spans_"
kwargs.setdefault("attr", f"{attr_prefix}{self.key}") kwargs.setdefault("attr", f"{attr_prefix}{self.key}")
kwargs.setdefault("labels", self.labels) kwargs.setdefault("allow_overlap", True)
kwargs.setdefault("multi_label", True)
kwargs.setdefault("threshold", self.cfg["threshold"])
kwargs.setdefault( kwargs.setdefault(
"getter", lambda doc, key: doc.spans.get(key[len(attr_prefix) :], []) "getter", lambda doc, key: doc.spans.get(key[len(attr_prefix) :], [])
) )