Assignment NameError: name 'lr_schedule' is not defined

Hi, in the C4_W2 Assignment I am getting NameError: name ‘lr_schedule’ is not defined, when training the model with callbacks=[lr_schedule]. What am I doing wrong? (Thanks for your hint!)

[code removed - moderator]

lr_schedule scope lies within the function create_model and so is undefined to the caller of this method.