Change conllu.TokenList to conllu.models.TokenList

combo/data/api.py

@@ -54,14 +54,14 @@ class Sentence:
         return len(self.tokens)
 
 
-class _TokenList(conllu.TokenList):
+class _TokenList(conllu.models.TokenList):
 
     @overrides
     def __repr__(self):
         return 'TokenList<' + ', '.join(token['token'] for token in self) + '>'
 
 
-def sentence2conllu(sentence: Sentence, keep_semrel: bool = True) -> conllu.TokenList:
+def sentence2conllu(sentence: Sentence, keep_semrel: bool = True) -> conllu.models.TokenList:
     tokens = []
     for token in sentence.tokens:
         token_dict = collections.OrderedDict(dataclasses.asdict(token))
@@ -82,7 +82,7 @@ def sentence2conllu(sentence: Sentence, keep_semrel: bool = True) -> conllu.Toke
                       metadata=sentence.metadata)
 
 
-def tokens2conllu(tokens: List[str]) -> conllu.TokenList:
+def tokens2conllu(tokens: List[str]) -> conllu.models.TokenList:
     return _TokenList(
         [collections.OrderedDict({"id": idx, "token": token}) for
          idx, token