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]>

Author: schmidt-ai

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

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",