diff --git a/test/requirements.txt b/test/requirements.txt
index 5b32f3d68f9d..9b17427fe9a5 100644
--- a/test/requirements.txt
+++ b/test/requirements.txt
@@ -9,7 +9,7 @@ pytest-rerunfailures
 pytest-timeout
 pytest-json-report
 pytest-xdist
-sagemaker>=2,<3
+sagemaker>=3.17.0,<4
 xmltodict
 boto3>=1.33
 retrying