03_training_model_on_tabular_data.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/prep_data/tabular_data/03_training_model_on_tabular_data.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [1]":
---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
in
1 import os
2 import sys
----> 3 import plotly.express as px
4 import plotly.offline as pyo
5 import plotly.graph_objs as go

ModuleNotFoundError: No module named 'plotly'