It is running in google colab in Python 3.10 with latest package versions. Are you sure its not a dependency issue on your end?