fix(rust, python): Set correct dtype for .extract_groups()

fix(rust, python): Set correct dtype for .extract_groups() #3163