Convert fea to feats in hanlp_common.document.Document.to_conll

Author: hankcs

plugins/hanlp_common/hanlp_common/document.py

@@ -129,7 +129,7 @@ def to_dict(self):
     def __str__(self) -> str:
         return self.to_json()
 
-    def to_conll(self, tok='tok', lem='lem', pos='pos', dep='dep', sdp='sdp') -> Union[
+    def to_conll(self, tok='tok', lem='lem', pos='pos', fea='fea', dep='dep', sdp='sdp') -> Union[
         CoNLLSentence, List[CoNLLSentence]]:
         """
         Convert to :class:`~hanlp_common.conll.CoNLLSentence`.
@@ -138,6 +138,7 @@ def to_conll(self, tok='tok', lem='lem', pos='pos', dep='dep', sdp='sdp') -> Uni
             tok (str): Field name for tok.
             lem (str): Field name for lem.
             pos (str): Filed name for upos.
+            fea (str): Field name for feats.
             dep (str): Field name for dependency parsing.
             sdp (str): Field name for semantic dependency parsing.
 
@@ -148,6 +149,7 @@ def to_conll(self, tok='tok', lem='lem', pos='pos', dep='dep', sdp='sdp') -> Uni
         tok = prefix_match(tok, self)
         lem = prefix_match(lem, self)
         pos = prefix_match(pos, self)
+        fea = prefix_match(fea, self)
         dep = prefix_match(dep, self)
         sdp = prefix_match(sdp, self)
         results = CoNLLSentenceList()
@@ -173,7 +175,8 @@ def get(_k, _i):
                 if not _dep:
                     _dep = (None, None)
                 sent.append(
-                    CoNLLUWord(i + 1, form=_tok, lemma=get(lem, i), upos=get(pos, i), head=_dep[0], deprel=_dep[1],
+                    CoNLLUWord(i + 1, form=_tok, lemma=get(lem, i), upos=get(pos, i), feats=get(fea, i), head=_dep[0],
+                               deprel=_dep[1],
                                deps=None if not get(sdp, i) else '|'.join(f'{x[0]}:{x[1]}' for x in get(sdp, i))))
             results.append(sent)
         if flat: