Any way to reorganize the distill heatmap

[robcli]

Hi,

My lab's been using DRAM for a little bit and it has been incredibly helpful. I was just wondering if there was anyway to reorganize the heatmap based on anything besides the fasta. I've tried adding columns to the annotations tsv and using the --groupby_column argument but I'm getting KeyError during the distill step. We want to be able to have a heatmap of MAGs from different groups and organize by their taxonomy.

Thanks