From c038626d52f68a1bc056f0052c042c6c8984e500 Mon Sep 17 00:00:00 2001
From: Jake Schmidt <132930658+schmidt-ai@users.noreply.github.com>
Date: Thu, 2 Nov 2023 09:43:51 -0600
Subject: [PATCH] Add `aws_batch` dependency extra (#773)

* create `aws_batch` extra

Signed-off-by: Jake Schmidt <jacob.schmidt@noetik.ai>

* unpin `boto3` in `dev-requirements.txt`

* unpin `aiobotocore` in `dev-requirements.txt`

Signed-off-by: Jake Schmidt <jacob.schmidt@noetik.ai>

---------

Signed-off-by: Jake Schmidt <jacob.schmidt@noetik.ai>
---
 dev-requirements.txt | 4 ++--
 setup.py             | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/dev-requirements.txt b/dev-requirements.txt
index d12f84012..5213954d7 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -1,7 +1,7 @@
-aiobotocore==2.4.2
+aiobotocore
 ax-platform[mysql]==0.2.3
 black==23.3.0
-boto3==1.24.59
+boto3
 captum>=0.4.0
 docker
 flake8==3.9.0
diff --git a/setup.py b/setup.py
index 0cd745121..3a37b67b7 100644
--- a/setup.py
+++ b/setup.py
@@ -81,6 +81,7 @@ def get_nightly_version():
             ],
         },
         extras_require={
+            "aws_batch": ["boto3"],
             "gcp_batch": [
                 "google-cloud-batch>=0.5.0",
                 "google-cloud-logging>=3.0.0",