Ok, it seems like you are using the 2020 version of the course. The current version uses a different plot_function which I was refering to (and which has the y_limit parameter), maybe there was a mixup. You can find the current course here and the function in particular in notebook 04. Does this solve you problem? 
3 Likes