I'm getting an error running this. Seems to work fine on Python 3.7, but on 3.8, I get the following.

Profiling your code with Flamegraph is simple. In Jupyter notebooks, click the 🔥 button above the cell you want to profile: For Python scripts, select Flamegraph: Profile file with py-spy from the ...