The example script build_model_from_scratch.py works only in ipython or jupyter

[abhisrkckl]

The script doesn't work if run from a command line or from a python REPL.

The issue is that there is a display() function used in this script that is undefined if run with python, but gets magically defined if run in jupyter or ipython.

The solution is to import it from IPython like this:
from IPython.core.display_functions import display