Upgrade SM Training toolkit package to accomodate latest fixes

satishpasumarthi · awsbmillare · commit 26b3fe4c54a5 · 2022-02-23T17:17:33.000-05:00
diff --git a/requirements.txt b/requirements.txt
@@ -14,7 +14,7 @@ sagemaker-inference==1.2.0
 retrying==1.3.3
 sagemaker-containers==2.8.6.post2
 sagemaker-inference==1.2.0
-sagemaker-training==3.6.2
+sagemaker-training==4.0.1
 scikit-learn==0.23.2
 scipy==1.5.3
 six==1.15.0
