There are only two types at the moment and we can therefore use a boolean. ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=203937357
