Merge pull request yzhangcs#1 from yzhangcs/main

yzhangcs · yzhangcs · commit 37f5f6ecde9d · 2021-03-10T09:46:17.000+09:00
Catch up 3 commits
diff --git a/supar/utils/field.py b/supar/utils/field.py
@@ -104,7 +104,7 @@ def __repr__(self):
         return s
 
     def __getstate__(self):
-        state = self.__dict__
+        state = dict(self.__dict__)
         if self.tokenize is None:
             state['tokenize_args'] = None
         elif self.tokenize.__module__.startswith('transformers'):
diff --git a/supar/utils/transform.py b/supar/utils/transform.py
@@ -501,12 +501,12 @@ def totree(cls, tokens, root=''):
 
         Examples:
             >>> print(Tree.totree(['She', 'enjoys', 'playing', 'tennis', '.'], 'TOP'))
-            (TOP (_ She) (_ enjoys) (_ playing) (_ tennis) (_ .))
+            (TOP ( (_ She)) ( (_ enjoys)) ( (_ playing)) ( (_ tennis)) ( (_ .)))
         """
 
         if isinstance(tokens[0], str):
             tokens = [(token, '_') for token in tokens]
-        tree = ' '.join([f"({pos} {word})" for word, pos in tokens])
+        tree = ' '.join([f"( ({pos} {word}))" for word, pos in tokens])
         return nltk.Tree.fromstring(f"({root} {tree})")
 
     @classmethod
