diff --git a/requirements.txt b/requirements.txt
index 622793f..f027524 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,9 +12,7 @@ psutil==5.7.2
 python-dateutil==2.8.1
 sagemaker-inference==1.2.0
 retrying==1.3.3
-sagemaker-containers==2.8.6.post2
-sagemaker-inference==1.2.0
-sagemaker-training==4.0.1
+sagemaker-training==4.1.2
 scikit-learn==1.0.2
 scipy==1.5.3
 six==1.15.0