It's not great that the example on the documentation home page does not work. I tried running that example (copy-pasted exactly from the home page) in a Google Colab notebook (after installing ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results
Feedback