House Price Interpretation Tutorial notebook fails

[jimthompson5802]

🐛 Bug

Running the tutorial Interpret regression models using Boston House Prices Dataset results in an IndexError in the last cell

To Reproduce

Steps to reproduce the behavior:

1. This colab notebook demonstrates the error.
2. pip install captum in the notebook
3. uploaded into the notebook container the pre-trained model.

IndexError                                Traceback (most recent call last)
<ipython-input-19-fd2cf6250225> in <module>()
      1 plt.figure(figsize=(15, 8))
      2 
----> 3 x_axis_data = np.arange(lc_attr_test.shape[1])
      4 
      5 y_axis_lc_attr_test = lc_attr_test.mean(0).detach().numpy()

IndexError: tuple index out of range
SEARCH STACK OVERFLOW
<Figure size 1080x576 with 0 Axes>

Expected behavior

Error free run of the tutorial notebook.

Environment

Google's colab notebook

 - Captum / PyTorch Version (0.5.0 / 1.12.1+cu113):
 - OS (e.g., Linux):
 - How you installed Captum / PyTorch (`conda`, `pip`, source): pip install captum
 - Build command you used (if compiling from source): N/A
 - Python version: 3.7.13
 - CUDA/cuDNN version: N/A
 - GPU models and configuration: colab GPU
 - Any other relevant information:  Error occurs with or without use of GPU

## Additional context

None.