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 * unpin `boto3` in `dev-requirements.txt` * unpin `aiobotocore` in `dev-requirements.txt` Signed-off-by: Jake Schmidt --------- Signed-off-by: Jake Schmidt --- 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",