from typing_extensions import Literal

def returns_a_or_b():
  # type: () -> Literal['a', 'b']
  return 'a'

asdf = returns_a_or_b()
if asdf == 'c':
    print("hey")