diff --git a/pyproject.toml b/pyproject.toml
index e735d3c52..7ac3f3cd9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -59,6 +59,7 @@ docker = {version = ">= 6.1.3", optional = true }
 kfp = { version = "2.3.0", optional = true, extras =["kubernetes"] }
 google-cloud-aiplatform =  { version = "1.34.0", optional = true}
 sagemaker = {version = ">= 2.197.0", optional = true}
+boto3 = {version = "1.28.64", optional = true}
 
 [tool.poetry.extras]
 component = ["dask"]
@@ -69,7 +70,7 @@ gcp = ["gcsfs"]
 
 kfp = ["docker", "kfp"]
 vertex = ["docker", "kfp", "google-cloud-aiplatform"]
-sagemaker = ["sagemaker"]
+sagemaker = ["sagemaker", "boto3"]
 docker = ["docker"]
 
 [tool.poetry.group.test.dependencies]