Function' object has no attribute 'shape

[balaji.ambresh]

Please follow these steps to refresh the workspace and fetch the starter code. Use it as a reference to fix undefined references in your notebook.