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

aws · Aug 13, 2024 · 0d7818a · 0d7818a
1 parent 36078a9
commit 0d7818a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 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

diff --git a/test/resources/versions/train.py b/test/resources/versions/train.py
@@ -21,8 +21,8 @@
 pyarrow==14.0.1
 python-dateutil==2.8.1
 retrying==1.3.3
-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
 smdebug==1.0.29