I am using Python in google Colab, I want to run the code from the github repository (https://github.com/dbgannon/NNets-and-Diffeqns/blob/master/burgernet.py)