mxnet_mnist_horovod.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/sagemaker-python-sdk/mxnet_horovod_mnist/mxnet_mnist_horovod.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [2]":
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
in
3 datasets.MNIST('data', download=True, transform=transforms.Compose([
4 transforms.ToTensor(),
----> 5 transforms.Normalize((0.1307,), (0.3081,))
6 ]))

TypeError: __init__() got an unexpected keyword argument 'download'