in grok/data.py [0:0]
def __init__(self, data_dir=DEFAULT_DATA_DIR) -> None:
self.token_file = bf.join(data_dir, self.token_file)
self.itos = self.get_tokens()
self.stoi: Dict[str, int] = dict([(s, i) for i, s in enumerate(self.itos)])