take a dependency on sagemaker-training and sagemaker-inference libs

aws · Aug 13, 2024 · 0e3761d · 0e3761d
1 parent 36078a9
commit 0e3761d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -18,8 +18,8 @@ pynvml==11.4.1
 python-dateutil==2.8.1
 retrying==1.3.3
 requests==2.29.0
-sagemaker-containers==2.8.6.post2
-sagemaker-inference==1.5.5
+sagemaker-training==4.7.4
+sagemaker-inference==1.10.1
 scikit-learn==0.24.1
 scipy==1.8.0
 urllib3==1.26.5