code: func (pk PrimaryKey) PartitionKeySet() map[string]struct{} {
  	m := make(map[string]struct{})
  	for _, p := range pk.PartitionKeys {
  		m[p] = struct{}{}
  	}
  	return m
  }
docs: // PartitionKeySet returns the set of partition keys