diff --git a/autogluon/inference/buildspec.yml b/autogluon/inference/buildspec.yml
index 003e4f9928d4..5320d394b3d9 100644
--- a/autogluon/inference/buildspec.yml
+++ b/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
 
diff --git a/autogluon/inference/docker/1.3/py3/Dockerfile.cpu b/autogluon/inference/docker/1.3/py3/Dockerfile.cpu
index 1389bc71c2a0..2a7b9ba77a32 100644
--- a/autogluon/inference/docker/1.3/py3/Dockerfile.cpu
+++ b/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
diff --git a/autogluon/inference/docker/1.3/py3/cu124/Dockerfile.gpu b/autogluon/inference/docker/1.3/py3/cu124/Dockerfile.gpu
index 1b1d52fb37fc..2b6400a43cbf 100644
--- a/autogluon/inference/docker/1.3/py3/cu124/Dockerfile.gpu
+++ b/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
diff --git a/autogluon/training/buildspec.yml b/autogluon/training/buildspec.yml
index 389b4f9015eb..7ace14ce0127 100644
--- a/autogluon/training/buildspec.yml
+++ b/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
 
diff --git a/autogluon/training/docker/1.3/py3/Dockerfile.cpu b/autogluon/training/docker/1.3/py3/Dockerfile.cpu
index 04102a0ab9b7..d93f07b61f05 100644
--- a/autogluon/training/docker/1.3/py3/Dockerfile.cpu
+++ b/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
diff --git a/autogluon/training/docker/1.3/py3/cu124/Dockerfile.gpu b/autogluon/training/docker/1.3/py3/cu124/Dockerfile.gpu
index 8d3fad5076c7..7b09d31596ea 100644
--- a/autogluon/training/docker/1.3/py3/cu124/Dockerfile.gpu
+++ b/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
diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml
index cfb8523f6275..8710cff73abb 100644
--- a/dlc_developer_config.toml
+++ b/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 = ""
diff --git a/src/patch_helper.py b/src/patch_helper.py
index d578375056a1..7ceb4a264151 100644
--- a/src/patch_helper.py
+++ b/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)
 
diff --git a/src/utils.py b/src/utils.py
index 9d8d2b6ec079..edeabafc23b0 100644
--- a/src/utils.py
+++ b/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
 
diff --git a/test/sagemaker_tests/autogluon/inference/resources/model/model_1.3.1.tar.gz b/test/sagemaker_tests/autogluon/inference/resources/model/model_1.3.1.tar.gz
new file mode 100644
index 000000000000..ebda4204bef3
Binary files /dev/null and b/test/sagemaker_tests/autogluon/inference/resources/model/model_1.3.1.tar.gz differ