How to remove branching logic in Python

ifelse.py

"""
  Before
"""

...
if lex_token['id']=='INT':
  yield LiteralToken(lex_token['value'])
elif lex_token['id']=='ADD_OP':
  yield OperatorAddToken()
elif lex_token['id']=='SUB_OP':
  yield OperatorSubToken()
elif lex_token['id']=='MUL_OP':
  yield OperatorMulToken()
elif lex_token['id']=='DIV_OP':
  yield OperatorDivToken()
elif lex_token['id']=='POW_OP':
  yield OperatorPowToken()
...

"""
  After
"""

tokenize_values = {
  'INT': LiteralToken,
  'ADD_OP': OperatorAddToken
}

def tokenize(program):
  for lex_token in lex_tokens:
    tokenize_values[lex_token['id']](lex_token['value'])
    
  ...