**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