from enum import IntEnum

class GraphIR(IntEnum):
    identity = 0
    relay = 1
    treelite = 2