diff --git a/requirements.txt b/requirements.txt
index 464d5b1..1f431a6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
-sagemaker>=1.71.0
+sagemaker>=1.71.0,<2.0.0
 boto3>=1.9.213
 pyyaml
diff --git a/setup.py b/setup.py
index 5476bfc..eb6c7fd 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@ def read_version():
 
 # Declare minimal set for installation
 required_packages = [
-    "sagemaker>=1.42.8",
+    "sagemaker>=1.71.0,<2.0.0",
     "boto3>=1.9.213",
     "pyyaml"
 ]