How do I get corrected data for target_condition using scArches?

[tjiyyanki1]

I noticed you provided a function to get batch resolved decoded dataset in your example here https://nbviewer.org/github/theislab/trVAE/blob/master/examples/trVAE_Haber.ipynb.

code: corrected_data = network.predict(adata,condition_key,target_condition=target_condition)

Is there a similar implementation in your scArches package?

Thanks