def __init__(self, branch: Branch, token_id: int):
    self.branch = branch
    self.token_id = token_id
def __init__(self, branch: Branch, token_id: int):
    self.branch = branch
    self.token_id = token_id