class Seahorse::Model::Shapes::UnionShape

def add_member_subclass(member, subclass)
  @member_subclasses[member] = subclass
end

def initialize(options = {})
  @member_subclasses = {}
  super options.merge(union: true)
end

def member_subclass(member)
  @member_subclasses[member]
end