class Dry::Schema::DSL

def key_map(types = self.types)
  keys = types.map { |key, type| key_spec(key, type) }
  km = KeyMap.new(keys)
  if key_map_type
    km.public_send(key_map_type)
  else
    km
  end
end