Skip to content
This repository was archived by the owner on Jan 15, 2024. It is now read-only.

Commit e746ba1

Browse files
authored
Improve Vocab.__repr__ if reserved_tokens or unknown_token is None (#989)
1 parent 7c4fd12 commit e746ba1

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/gluonnlp/vocab/vocab.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -512,8 +512,9 @@ def __call__(self, tokens):
512512
return self[tokens]
513513

514514
def __repr__(self):
515-
return 'Vocab(size={}, unk="{}", reserved="{}")'.format(len(self), self._unknown_token,
516-
self._reserved_tokens)
515+
unk = '"{}"'.format(self._unknown_token) if self._unknown_token else 'None'
516+
reserved = '"{}"'.format(self._reserved_tokens) if self._reserved_tokens else 'None'
517+
return 'Vocab(size={}, unk={}, reserved={})'.format(len(self), unk, reserved)
517518

518519
def to_json(self):
519520
"""Serialize Vocab object to json string.

0 commit comments

Comments
 (0)