diff --git a/build_artifacts/v2/v2.0/v2.0.0/cpu.additional_packages_env.in b/build_artifacts/v2/v2.0/v2.0.0/cpu.additional_packages_env.in
index 7495d996..53d1aa44 100644
--- a/build_artifacts/v2/v2.0/v2.0.0/cpu.additional_packages_env.in
+++ b/build_artifacts/v2/v2.0/v2.0.0/cpu.additional_packages_env.in
@@ -1,3 +1,4 @@
 #Add any additional packages here
 conda-forge::mlflow[version='>=2.13.2,<3.0']
 conda-forge::sagemaker-mlflow[version='>=0.1.0,<1.0']
+conda-forge::jupyter-idle[version'>=0.1.1,<1.0']
diff --git a/build_artifacts/v2/v2.0/v2.0.0/gpu.additional_packages_env.in b/build_artifacts/v2/v2.0/v2.0.0/gpu.additional_packages_env.in
index 7495d996..53d1aa44 100644
--- a/build_artifacts/v2/v2.0/v2.0.0/gpu.additional_packages_env.in
+++ b/build_artifacts/v2/v2.0/v2.0.0/gpu.additional_packages_env.in
@@ -1,3 +1,4 @@
 #Add any additional packages here
 conda-forge::mlflow[version='>=2.13.2,<3.0']
 conda-forge::sagemaker-mlflow[version='>=0.1.0,<1.0']
+conda-forge::jupyter-idle[version'>=0.1.1,<1.0']
diff --git a/test/test_artifacts/v2/jupyter-idle-ext.test.Dockerfile b/test/test_artifacts/v2/jupyter-idle-ext.test.Dockerfile
new file mode 100644
index 00000000..26c4e055
--- /dev/null
+++ b/test/test_artifacts/v2/jupyter-idle-ext.test.Dockerfile
@@ -0,0 +1,6 @@
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+CMD ["python", "-c", "import jupyter_idle"]
diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py
index 43e1ca99..a66e8793 100644
--- a/test/test_dockerfile_based_harness.py
+++ b/test/test_dockerfile_based_harness.py
@@ -45,6 +45,7 @@
         ("serve.test.Dockerfile", ["serve-langchain"]),
         ("langchain-aws.test.Dockerfile", ["langchain-aws"]),
         ("mlflow.test.Dockerfile", ["mlflow"]),
+        ("jupyter-idle-ext.test.Dockerfile", ["jupyter-idle"]),
     ],
 )
 def test_dockerfiles_for_cpu(
@@ -86,6 +87,7 @@ def test_dockerfiles_for_cpu(
         ("langchain-aws.test.Dockerfile", ["langchain-aws"]),
         ("mlflow.test.Dockerfile", ["mlflow"]),
         ("sagemaker-mlflow.test.Dockerfile", ["sagemaker-mlflow"]),
+        ("jupyter-idle-ext.test.Dockerfile", ["jupyter-idle"]),
     ],
 )
 def test_dockerfiles_for_gpu(