in client/sql/dataframe.go [195:211]
func (df *dataFrameImpl) Repartition(numPartitions int, columns []string) (DataFrame, error) {
var partitionExpressions []*proto.Expression
if columns != nil {
partitionExpressions = make([]*proto.Expression, 0, len(columns))
for _, c := range columns {
expr := &proto.Expression{
ExprType: &proto.Expression_UnresolvedAttribute_{
UnresolvedAttribute: &proto.Expression_UnresolvedAttribute{
UnparsedIdentifier: c,
},
},
}
partitionExpressions = append(partitionExpressions, expr)
}
}
return df.repartitionByExpressions(numPartitions, partitionExpressions)
}