Custom Weights Initializer for layer weights using any Tensorflow Probability Distribution

Question

I want to create a custom TF layer and initialize the layer weights with a distribution from TFP. But I am running into the following problem. For reference, I subclassed th