explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-27 17:54:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

use field validator

Browse Source
This commit is contained in:
India Kerle 2024-03-04 09:53:53 -03:00
parent 3b37fb6dcf
commit 59d8ee4132
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/coordinationruler.py
View File

@ -1,7 +1,7 @@
import re import re
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
from pydantic import BaseModel, validator from pydantic import BaseModel, field_validator
from ..language import Language from ..language import Language
from ..tokens import Doc, Token from ..tokens import Doc, Token
@ -137,7 +137,7 @@ def split_noun_coordination(doc: Doc) -> Union[List[str], None]:
class SplittingRule(BaseModel): class SplittingRule(BaseModel):
function: Callable[[Doc], Union[List[str], None]] function: Callable[[Doc], Union[List[str], None]]
@validator("function") @field_validator("function")
def check_return_type(cls, v): def check_return_type(cls, v):
dummy_doc = Doc(Language().vocab, words=["dummy", "doc"], spaces=[True, False]) dummy_doc = Doc(Language().vocab, words=["dummy", "doc"], spaces=[True, False])
result = v(dummy_doc) result = v(dummy_doc)