Ok so this error appears when I import first tensorflow.compat.v1 and than import also tensorflow_probability. It only appear when I run this twice (even if there is not other code involved).
I can’t reproduce this error in colab so I guess this error has to do with some clash in packages on my laptop.