Issues running the updated version of MrVI

Help scvi-tools
1

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

2

Hi @shani.bmo,
Thanks for trying the new version. This was an issue with a jax update that we should have fixed. Could you try again with the implementation in scvi-tools if you haven’t already?