Scikit-learn Estimator Example With Batch Transform.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/aws_sagemaker_studio/sagemaker-python-sdk/scikit_learn_iris/Scikit-learn Estimator Example With Batch Transform.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [4]":
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
in
8 role=role,
9 sagemaker_session=sagemaker_session,
---> 10 hyperparameters={'max_leaf_nodes': 30})

/opt/conda/lib/python3.7/site-packages/sagemaker/sklearn/estimator.py in __init__(self, entry_point, framework_version, py_version, source_dir, hyperparameters, image_uri, **kwargs)
115 "train_instance_count", "instance_count", kwargs.get("instance_count"), kwargs
116 )
--> 117 validate_version_or_image_args(framework_version, py_version, image_uri)
118 if py_version and py

[...]

ValueError: framework_version or py_version was None, yet image_uri was also None. Either specify both framework_version and py_version, or specify image_uri.