1.3.1 Test

autogluon/inference/buildspec.yml

@@ -1,7 +1,7 @@
 account_id: &ACCOUNT_ID <set-$ACCOUNT_ID-in-environment>
 region: &REGION <set-$REGION-in-environment>
 framework: &FRAMEWORK autogluon
-version: &VERSION 1.3.0
+version: &VERSION 1.3.1
 short_version: &SHORT_VERSION 1.3
 arch_type: x86
 

autogluon/inference/docker/1.3/py3/Dockerfile.cpu

@@ -19,7 +19,7 @@ RUN apt-get update \
  && apt-get clean \
  && rm -rf /var/lib/apt/lists/*
 
-ARG AUTOGLUON_VERSION=1.3.0
+ARG AUTOGLUON_VERSION=1.3.1
 
 # Upgrading pip and installing/updating Python dependencies
 # Comments are added to explain the reason behind each update

autogluon/inference/docker/1.3/py3/cu124/Dockerfile.gpu

@@ -20,7 +20,7 @@ RUN apt-get update \
  && apt-get clean \
  && rm -rf /var/lib/apt/lists/*
 
-ARG AUTOGLUON_VERSION=1.3.0
+ARG AUTOGLUON_VERSION=1.3.1
 
 
 # Upgrading pip and installing/updating Python dependencies

autogluon/training/buildspec.yml

@@ -1,7 +1,7 @@
 account_id: &ACCOUNT_ID <set-$ACCOUNT_ID-in-environment>
 region: &REGION <set-$REGION-in-environment>
 framework: &FRAMEWORK autogluon
-version: &VERSION 1.3.0
+version: &VERSION 1.3.1
 short_version: &SHORT_VERSION 1.3
 arch_type: x86
 

autogluon/training/docker/1.3/py3/Dockerfile.cpu

@@ -20,7 +20,7 @@ RUN apt-get update \
  && apt-get clean \
  && rm -rf /var/lib/apt/lists/*
 
-ARG AUTOGLUON_VERSION=1.3.0
+ARG AUTOGLUON_VERSION=1.3.1
 
 # Upgrading pip and installing/updating Python dependencies
 # Comments are added to explain the reason behind each update

autogluon/training/docker/1.3/py3/cu124/Dockerfile.gpu

@@ -18,7 +18,7 @@ RUN apt-get update \
  && rm -rf /var/lib/apt/lists/* \
  && rm -rf /tmp/*
 
-ARG AUTOGLUON_VERSION=1.3.0
+ARG AUTOGLUON_VERSION=1.3.1
 
 
 # Upgrading pip and installing/updating Python dependencies

dlc_developer_config.toml

@@ -37,7 +37,8 @@ deep_canary_mode = false
 [build]
 # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
 # available frameworks - ["base", "vllm", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
-build_frameworks = []
+build_frameworks=["autogluon"]
+
 
 
 # By default we build both training and inference containers. Set true/false values to determine which to build.
@@ -121,7 +122,8 @@ use_scheduler = false
 # Standard Framework Training
 dlc-pr-pytorch-training = ""
 dlc-pr-tensorflow-2-training = ""
-dlc-pr-autogluon-training = ""
+dlc-pr-autogluon-training = "autogluon/training/buildspec.yml"
+
 
 # ARM64 Training
 dlc-pr-pytorch-arm64-training = ""
@@ -151,7 +153,8 @@ dlc-pr-tensorflow-2-habana-training = ""
 # Standard Framework Inference
 dlc-pr-pytorch-inference = ""
 dlc-pr-tensorflow-2-inference = ""
-dlc-pr-autogluon-inference = ""
+dlc-pr-autogluon-inference = "autogluon/inference/buildspec.yml"
+
 
 # Graviton Inference
 dlc-pr-pytorch-graviton-inference = ""

src/patch_helper.py

@@ -24,7 +24,7 @@
 from codebuild_environment import get_cloned_folder_path
 from context import Context
 
-from src.constants import PATCHING_INFO_PATH_WITHIN_DLC
+from constants import PATCHING_INFO_PATH_WITHIN_DLC
 
 FORMATTER = OutputFormatter(constants.PADDING)
 

src/utils.py

@@ -18,6 +18,10 @@
 import json
 import logging
 import sys
+
+_project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+if _project_root not in sys.path:
+    sys.path.insert(0, _project_root)
 import boto3
 import constants