[python] Fix typo in embedding notebook

[ivirshup]

Fixes #1206

I've re-run the notebook locally, so don't think I've introduced a new error