Cleanup: Add back images to patches, comment out autoPR queue calls (#…

…4262)
aws · Sep 19, 2024 · 27931d3 · 27931d3
1 parent 3c5e72d
commit 27931d3
Show file tree

Hide file tree

Showing 3 changed files with 63 additions and 63 deletions.
diff --git a/pytorch/inference/docker/2.3/py3/cu121/Dockerfile.gpu b/pytorch/inference/docker/2.3/py3/cu121/Dockerfile.gpu
@@ -234,7 +234,6 @@ RUN pip install --no-cache-dir -U \
     ${TORCHTEXT_URL}
 
 # Install TorchServe pypi dependencies directly from their requirements.txt file
-# NOTE: This also brings in unnecessary cpu dependencies like nvgpu
 RUN pip install --no-cache-dir -U -r https://raw.githubusercontent.com/pytorch/serve/v${TORCHSERVE_VERSION}/requirements/common.txt \
  && pip install --no-cache-dir -U \
     torchserve==${TORCHSERVE_VERSION} \

diff --git a/release_images_patches.yml b/release_images_patches.yml
@@ -108,33 +108,33 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  # 9:
-  #   framework: "pytorch"
-  #   version: "2.1.0"
-  #   arch_type: "graviton"
-  #   customer_type: "ec2"
-  #   inference:
-  #     device_types: [ "cpu" ]
-  #     python_versions: [ "py310" ]
-  #     os_version: "ubuntu20.04"
-  #     example: False
-  #     disable_sm_tag: False
-  #     force_release: False
-  # 10:
-  #   framework: "pytorch"
-  #   version: "2.1.0"
-  #   arch_type: "graviton"
-  #   customer_type: "sagemaker"
-  #   inference:
-  #     device_types: [ "cpu" ]
-  #     python_versions: [ "py310" ]
-  #     os_version: "ubuntu20.04"
-  #     example: False
-  #     disable_sm_tag: False
-  #     force_release: False
+  7:
+    framework: "pytorch"
+    version: "2.1.0"
+    arch_type: "graviton"
+    customer_type: "ec2"
+    inference:
+      device_types: [ "cpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu20.04"
+      example: False
+      disable_sm_tag: False
+      force_release: False
+  8:
+    framework: "pytorch"
+    version: "2.1.0"
+    arch_type: "graviton"
+    customer_type: "sagemaker"
+    inference:
+      device_types: [ "cpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu20.04"
+      example: False
+      disable_sm_tag: False
+      force_release: False
 
   # PyTorch 2.2 Images
-  7:
+  9:
     framework: "pytorch"
     version: "2.2.0"
     arch_type: "x86"
@@ -147,7 +147,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  8:
+  10:
     framework: "pytorch"
     version: "2.2.0"
     arch_type: "x86"
@@ -173,7 +173,7 @@ release_images:
   #     example: False
   #     disable_sm_tag: False
   #     force_release: False
-  9:
+  11:
     framework: "pytorch"
     version: "2.2.0"
     arch_type: "x86"
@@ -186,33 +186,33 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  # 15:
-  #   framework: "pytorch"
-  #   version: "2.2.1"
-  #   arch_type: "graviton"
-  #   customer_type: "ec2"
-  #   inference:
-  #     device_types: [ "cpu" ]
-  #     python_versions: [ "py310" ]
-  #     os_version: "ubuntu20.04"
-  #     example: False
-  #     disable_sm_tag: False
-  #     force_release: False
-  # 16:
-  #   framework: "pytorch"
-  #   version: "2.2.1"
-  #   arch_type: "graviton"
-  #   customer_type: "sagemaker"
-  #   inference:
-  #     device_types: [ "cpu" ]
-  #     python_versions: [ "py310" ]
-  #     os_version: "ubuntu20.04"
-  #     example: False
-  #     disable_sm_tag: False
-  #     force_release: False
+  12:
+    framework: "pytorch"
+    version: "2.2.1"
+    arch_type: "graviton"
+    customer_type: "ec2"
+    inference:
+      device_types: [ "cpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu20.04"
+      example: False
+      disable_sm_tag: False
+      force_release: False
+  13:
+    framework: "pytorch"
+    version: "2.2.1"
+    arch_type: "graviton"
+    customer_type: "sagemaker"
+    inference:
+      device_types: [ "cpu" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu20.04"
+      example: False
+      disable_sm_tag: False
+      force_release: False
 
   # PyTorch 2.3 Images
-  10:
+  14:
     framework: "pytorch"
     version: "2.3.0"
     customer_type: "ec2"
@@ -225,7 +225,7 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
-  11:
+  15:
     framework: "pytorch"
     version: "2.3.0"
     customer_type: "sagemaker"
@@ -251,7 +251,7 @@ release_images:
   #     example: False
   #     disable_sm_tag: False
   #     force_release: False
-  12:
+  16:
     framework: "pytorch"
     version: "2.3.0"
     arch_type: "x86"
@@ -303,7 +303,7 @@ release_images:
   #     example: False
   #     disable_sm_tag: False
   #     force_release: False
-  13:
+  17:
     framework: "tensorflow"
     version: "2.14.1"
     customer_type: "sagemaker"
@@ -383,7 +383,7 @@ release_images:
   #     example: False
   #     disable_sm_tag: False
   #     force_release: False
-  14:
+  18:
     framework: "tensorflow"
     version: "2.16.2"
     customer_type: "sagemaker"

diff --git a/src/autopr_caller.py b/src/autopr_caller.py
@@ -206,12 +206,13 @@ def main():
     LOGGER.info(
         f"Message body to be sent to AutoPR Queue: {json.dumps(message_body_to_be_sent_to_autopr_queue)}"
     )
-    if os.getenv("BUILD_CONTEXT") == "MAINLINE":
-        send_message_to_queue(
-            queue_name=AUTOPR_PROD_QUEUE,
-            queue_region=QUEUE_REGION,
-            message_body_string=json.dumps(message_body_to_be_sent_to_autopr_queue),
-        )
+    # TODO: Uncomment when feature is re-enabled
+    # if os.getenv("BUILD_CONTEXT") == "MAINLINE":
+    #     send_message_to_queue(
+    #         queue_name=AUTOPR_PROD_QUEUE,
+    #         queue_region=QUEUE_REGION,
+    #         message_body_string=json.dumps(message_body_to_be_sent_to_autopr_queue),
+    #     )
 
 
 if __name__ == "__main__":