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

Malav Shastri · Malav Shastri · commit 0d7818a32f45 · 2024-08-13T00:22:25.000-04:00
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
