create aws_batch extra

Signed-off-by: Jake Schmidt <[email protected]>
pytorch · Oct 10, 2023 · 49b4dae · 49b4dae
1 parent 13a4787
commit 49b4dae
Showing 1 changed file with 1 addition and 0 deletions.
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",