Release 1.0.4

combo/data/api.py

@@ -55,6 +55,7 @@ def sentence2conllu(sentence: Sentence, keep_semrel: bool = True) -> conllu.Toke
         # Remove semrel to have default conllu format.
         if not keep_semrel:
             del token_dict["semrel"]
+        del token_dict["embeddings"]
         tokens.append(token_dict)
     # Range tokens must be tuple not list, this is conllu library requirement
     for t in tokens:
@@ -84,10 +85,10 @@ def conllu2sentence(conllu_sentence: conllu.TokenList,
     if sentence_embedding is None:
         sentence_embedding = []
     tokens = []
-    for idx, token in enumerate(conllu_sentence.tokens):
+    for token in conllu_sentence.tokens:
         tokens.append(
             Token(
-                **token, embeddings=embeddings[idx]
+                **token, embeddings=embeddings[token["id"]]
             )
         )
     return Sentence(