I am having the same issue as many others and their solutions haven\'t worked.
Im using tensorflow 2.2.0 and tensorflow probability 0.10.0.
Package