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

Turn on formatting for allow_extra_label

Browse Source
This commit is contained in:
Lj Miranda 2022-11-29 10:56:11 +08:00
parent 4b5c5f197b
commit 270db33dcf
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spacy/pipeline/spancat.py
View File

@ -230,15 +230,17 @@ class SpanCategorizer(TrainablePipe):
def _allow_extra_label(self) -> None:
"""Raise an error if the component can not add any more labels."""
nO = None
# fmt: off
if self.model.has_dim("nO"):
nO = self.model.get_dim("nO")
elif self.model.has_ref("output_layer") and self.model.get_ref("output_layer").has_dim("nO"):
elif self.model.has_ref("output_layer") and self.model.get_ref(
"output_layer"
).has_dim("nO"):
nO = self.model.get_ref("output_layer").get_dim("nO")
if nO is not None and nO == self._n_labels:
if not self.is_resizable:
raise ValueError(Errors.E922.format(name=self.name, nO=self.model.get_dim("nO")))
# fmt: on
raise ValueError(
Errors.E922.format(name=self.name, nO=self.model.get_dim("nO"))
)
def add_label(self, label: str) -> int:
"""Add a new label to the pipe.