Add aws_batch dependency extra (#773)

* create `aws_batch` extra Signed-off-by: Jake Schmidt <[email protected]> * unpin `boto3` in `dev-requirements.txt` * unpin `aiobotocore` in `dev-requirements.txt` Signed-off-by: Jake Schmidt <[email protected]> --------- Signed-off-by: Jake Schmidt <[email protected]>
pytorch · Nov 2, 2023 · c038626 · c038626
1 parent 1fe0607
commit c038626
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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",