Issue found in Vector-Quantized Variational Autoencoders code example

[eng-amrahmed]

trafficstars

The below line throws an exception sampled = dist.sample()

I have solved the issue by installing the tensorFlow-probability version (0.12.0) but it's just a temporary fix.

//!pip install -q tensorflow-probability !pip install -q "tensorflow-probability==0.12.0"