GeneVsCountsScatter not responding to count_transform argument

[spficklin]

When running the following with raw counts that have been upper quantize transformed:

selected_genes = gsc.gene_sets["Treatment"].gene_support()[:50]
hv.output(backend='bokeh')
plot = gsf.plots.gem.GeneVsCountsScatter(gsc, selected_genes=selected_genes, count_variable='uq_counts',
                                         hue="Treatment", count_transform=lambda counts: np.log2(counts.where(counts > 0)))
plot

The following plot appears:

It seems the count_transform argument is not being applied.