diff --git a/aws/amazon-sagemaker/pytorch-mnist/poetry.lock b/aws/amazon-sagemaker/pytorch-mnist/poetry.lock
index 3dcb12e930..e42451e331 100644
--- a/aws/amazon-sagemaker/pytorch-mnist/poetry.lock
+++ b/aws/amazon-sagemaker/pytorch-mnist/poetry.lock
@@ -1364,13 +1364,13 @@ crt = ["botocore[crt] (>=1.33.2,<2.0a.0)"]
 
 [[package]]
 name = "sagemaker"
-version = "2.200.1"
+version = "2.201.0"
 description = "Open source library for training and deploying models on Amazon SageMaker."
 optional = false
 python-versions = ">= 3.8"
 files = [
-    {file = "sagemaker-2.200.1-py2.py3-none-any.whl", hash = "sha256:cc06d44f266268d0945384485ce08079c66b64dc8672da0198868452e9451a53"},
-    {file = "sagemaker-2.200.1.tar.gz", hash = "sha256:2fdc7495ad00a0f1c04b8373615026d37869ade20735d4b42c05bb188bf4fc4f"},
+    {file = "sagemaker-2.201.0-py2.py3-none-any.whl", hash = "sha256:c9ce160d5aec7ea6287239b74115aa04662c17c20c12912c306583fca74bb4d7"},
+    {file = "sagemaker-2.201.0.tar.gz", hash = "sha256:f7d74cc9e26172860ec9a90227973565fbd05de8d5452cbc226d40d778a43ce7"},
 ]
 
 [package.dependencies]
@@ -1731,4 +1731,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = "3.10.x"
-content-hash = "73143b8b083a27cba1d7ccfae8f0cb2fc264eab89e8325a897bfe320aeecf0a8"
+content-hash = "ee3609bdb2a55f48d48c8435c61e56b1339d4db650e8cd6bff3a69136a6f7251"
diff --git a/aws/amazon-sagemaker/pytorch-mnist/pyproject.toml b/aws/amazon-sagemaker/pytorch-mnist/pyproject.toml
index 8deda7b476..ec06130e6b 100644
--- a/aws/amazon-sagemaker/pytorch-mnist/pyproject.toml
+++ b/aws/amazon-sagemaker/pytorch-mnist/pyproject.toml
@@ -11,7 +11,7 @@ torchvision = "0.16.2"
 
 [tool.poetry.group.dev.dependencies]
 poethepoet = "0.24.4"
-sagemaker = "2.200.1"
+sagemaker = "2.201.0"
 
 [build-system]
 requires = ["poetry-core"]