Using categorical_covariate_keys when sampling or generating normalised expression

Hi,

thank you for you fast reply! I realised later that I am looking for conditional sampling from the prior (latent space) and then decoding with covariates for interest. I’ve found a similar question here and the answer is a bit outdated because scvi-tools has been updates since. But I guess the general setup is similar, we sample from the latent normal distribution and then we can now use scvi.module.vae.generative() to decode, is this correct?

Could you help with this?