From ee15bd68162c9ae9f3aac2bcd9a9a8b1b3323ca0 Mon Sep 17 00:00:00 2001
From: Stephen Mistele <smistele@amazon.com>
Date: Fri, 8 Aug 2025 13:34:04 -0700
Subject: [PATCH] feat: update sagemaker training to v5.1.0

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index dbdf9745..28dd59bf 100644
--- a/setup.py
+++ b/setup.py
@@ -74,7 +74,7 @@ def read_version():
         "Programming Language :: Python :: 3.9",
     ],
     install_requires=[
-        "sagemaker-training>=4.3.0,<=4.8.3",
+        "sagemaker-training>=4.3.0,<=5.1.0",
         "numpy < 1.24",
         "scipy",
         "scikit-learn",