diff --git a/tests/e2e/requirements.txt b/tests/e2e/requirements.txt
index ac9ae8fe5..ba6f85640 100644
--- a/tests/e2e/requirements.txt
+++ b/tests/e2e/requirements.txt
@@ -8,5 +8,5 @@ mysql-connector-python==8.0.33
 retrying==1.3.4
 requests==2.31.0
 numpy==1.24.2
-sagemaker==2.148.0
+sagemaker==2.218.0
 pandas==2.0.0
\ No newline at end of file