pyspark collect_set ou collect_list com groupby

Como posso usarcollect_set oucollect_list em um quadro de dados apósgroupby. por exemplo:df.groupby('key').collect_set('values'). Eu recebo um erro:AttributeError: 'GroupedData' object has no attribute 'collect_set'