Documentation example for UDE learning dynamics has broken

[mirams]

Describe the example

The documentation example here: Automatically Discover Missing Physics by Embedding Machine Learning into Differential Equations appears to have seriously regressed:

I guess regression tests on documentation would be worthwhile, there is something not being picked up by standard tests (#104 related)