Expose size attribute of PassThroughVocabulary

PiperOrigin-RevId: 638655043
google · May 30, 2024 · 4e09f56 · 4e09f56
1 parent 9f39c43
commit 4e09f56
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/seqio/vocabularies.py b/seqio/vocabularies.py
@@ -199,6 +199,10 @@ def eos_id(self) -> Optional[Any]:
   def unk_id(self) -> Optional[Any]:
     return None
 
+  @property
+  def size(self) -> int:
+    return self._size
+
   def __eq__(self, other):
     if not isinstance(other, PassThroughVocabulary):
       return False