diff --git a/sagemaker-train/pyproject.toml b/sagemaker-train/pyproject.toml
index c7fdb2237e..3382363588 100644
--- a/sagemaker-train/pyproject.toml
+++ b/sagemaker-train/pyproject.toml
@@ -44,7 +44,7 @@ dependencies = [
     "mlflow>=3.0.0,<4.0.0",
     "nest_asyncio>=1.5.0",
     "requests>=2.28.0",
-    "pydantic>=2.0.0",
+    "pydantic>=2.0.0,<3.0.0",
 ]
 
 [project.urls]