Dropping this issue here since it might be a bigger issue than just scikit learn container. I observed similar issue in xgboost container as well. https://github.com/aws/sagemaker-scikit-learn-container/issues/149