Using_Algorithm_Arn_From_AWS_Marketplace.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/aws_marketplace/using_algorithms/amazon_demo_product/Using_Algorithm_Arn_From_AWS_Marketplace.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [5]":
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
in
8 train_instance_count=1,
9 train_instance_type='ml.c4.xlarge',
---> 10 base_job_name='scikit-from-aws-marketplace')

TypeError: __init__() missing 2 required positional arguments: 'instance_count' and 'instance_type'