Issues running the updated version of MrVI

Hi all,
we would like to use the newer version of MrVI (1.0.0b1), but we can not run it -

when trying to train the model:

model = MrVI(adata)
model.train()

we get the follwing error:

TypeError: scvi.train._trainer.Trainer() got multiple values for keyword argument 'devices'

We suspect it’s a dependency issue (divices and other arguments are given twice, once explicitely as the train function defalut, and once through the **trainer_kwargs).

We would love to get your input on that - any idea how to fix it?

We would also be happy to get the dependecies list, to make sure we are using the same versions.

Thanks for your help,
Shani