Custom Distribution Help

Hi,

I am attempting to write a custom distribution but I can’t get it to work. I copied some examples from the internet and they don’t work, the trace command just hangs. I made a simple notebook to illustrate this.
If anyone can get this to run, then I know that the issue is my python environment. Right now, I have no way to know because there is no error message when it hangs. It just hangs.

Note: google colab is worse and better in some sense. You have to install packages to get it to work at all. But I don’t get the hanging error. I get a different error there (which is why I suspect it’s an environment issue).

Any thoughts are appreciated.