Add a __str__ to the CompositeVocab

stanfordnlp · Nov 21, 2023 · 7662830 · 7662830
1 parent d598368
commit 7662830
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/stanza/models/common/vocab.py b/stanza/models/common/vocab.py
@@ -202,6 +202,11 @@ def lens(self):
     def items(self, idx):
         return self._id2unit[idx]
 
+    def __str__(self):
+        pieces = ["[" + ",".join(x) + "]" for _, x in self._id2unit.items()]
+        rep = "<{}:\n {}>".format(type(self), "\n ".join(pieces))
+        return rep
+
 class BaseMultiVocab:
     """ A convenient vocab container that can store multiple BaseVocab instances, and support 
     safe serialization of all instances via state dicts. Each subclass of this base class