pyspark collect_set или collect_list с групповым

Как я могу использоватьcollect_set или жеcollect_list на кадре данных послеgroupby, например:df.groupby('key').collect_set('values'), Я получаю ошибку:AttributeError: 'GroupedData' object has no attribute 'collect_set'