diff --git a/pyproject.toml b/pyproject.toml
index d1c634f86d..1c60e786e2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -47,6 +47,7 @@ dependencies = [
   "psutil",
   "PyYAML~=6.0",
   "requests",
+  "sagemaker-core>=1.0.0,<2.0.0",
   "schema",
   "smdebug_rulesconfig==1.0.1",
   "tblib>=1.7.0,<4",