Toy example with Tensorflow-probability and a LambdaVariational layer as last layer with a Beta distribution fails

Question

I am using a toy example in an attempt to model the parameter of a Bernoulli distribution modeling a binary variable with a Beta distribution using tensorflow-probability bu