Error when visualizing forecasts

When running the notebook, say timesfm.ipynb, I get the following error when running `tfm.plot(val_dataset)` cell:

```
NameError: name 'context_len' is not defined
```

We need to fix the bug in the visualization function.