From bd343453aaebe5026fa404c91c16042876a77791 Mon Sep 17 00:00:00 2001
From: Georges Lorre <georges.lorre@ml6.eu>
Date: Wed, 22 Nov 2023 15:58:13 +0100
Subject: [PATCH] Add boto3 ass optinal dependency

---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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]