Building your own algorithm container scikit_build_your_own_container is breaking due to recent changes in the sagemaker SDK

[kirannsa]

Link to the notebook
Add the link to the notebook.

  build_and_train_models/sm-scikit_build_your_own_container/sm-scikit_build_your_own_container.ipynb
  
 https://github.com/aws/amazon-sagemaker-examples/blob/f85ce9ef18aeecdab23a07b80ec3b45463aa3b44/%20%20%20%20%20%20build_and_train_models/sm-scikit_build_your_own_container/sm-scikit_build_your_own_container.ipynb

What aspects of the notebook can be improved?
recent changes to the SDK

What are your suggestions?
predict functionality is failing