Adding sample IDs of those with non-ref GT as row variable

Hail Query & hailctl
1

I would like to add a row variable that contains sample ID(s) of hets or homozygous for the alt allele. Tried a few things but no success. Would greatly appreciate any ideas on how to gather this data

2

something like this:

mt = mt.annotate_rows(
  het_ids = hl.agg.filter(mt.GT.is_het(), hl.agg.collect(mt.s)),
  homvar_ids = hl.agg.filter(mt.GT.is_hom_var(), hl.agg.collect(mt.s)))
3

this worked! thanks so much for the speedy reply. Much appreciated