diff --git a/components/aws/sagemaker/requirements.txt b/components/aws/sagemaker/requirements.txt
index 470d09e177a..94790a05876 100644
--- a/components/aws/sagemaker/requirements.txt
+++ b/components/aws/sagemaker/requirements.txt
@@ -1,6 +1,6 @@
 # Runtime requirements
 boto3==1.14.12
-sagemaker==2.1.0
+sagemaker==2.256.0
 pathlib2==2.3.5
 pyyaml==5.4
 mypy-extensions==0.4.3