Batch correction with MOFA

[alitinet]

Hi,

I wanted to double check that I'm running MOFA correctly. I have two batches in my data and the function I'm using is

mu.tl.mofa(mdata, groups_label='batch', gpu_mode=True)

The resulting UMAP looks like this. Is this the correct way to run MOFA on batches? Thanks!