MultivariateNormalDiag is not in the TensorFlow version installed by default by installTF()

[msabvid]

Hi, I'm following the instructions from https://github.com/STOR-i/sgmcmc to install sgmcmc directly from your GitHub, and after trying to run the sgldcv for Bayesian regression, I get the following error:

Python error output:
Error in py_get_attr_impl(x, name, silent): AttributeError: module 'tensorflow._api.v1.distributions' has no attribute 'MultivariateNormalDiag'

Check your tensorflow code specifying the logLik and logPrior functions is correct.
Ensure constants in logLik and logPrior functions are specified as type float32 using
tf$constant(.., dtype = tf$float32) -- see the tutorials for some examples.