Dimensionality error in DREAM algorithm?

[pseur]

Thank you very much for developing this package. I would like to use the DREAM sampler to calibrate uncertain parameters (i.e., sample a posterior distribution) using new experimental data I collected. There, to update the state after the first sampling, I see that you are sampling a random number "norm_vars", which has shape "(self.n_chain, self.n_chain)".

To me however, it should be of dimension "(self.n_chains,current_state.shape[1])", as current_state.shape[1] is equal to the number of uncertain parameters to calibrate (i.e., the state).

Otherwise, it cannot be added to the entries of the state in the following line:

I hope it makes sense,
Thank you in advance!