feature: Add jupyter-idle extension (aws#453)

Co-authored-by: Clayton Parnell <[email protected]>

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-activity-monitor-extension[version'>=0.1.0,<1.0']

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-activity-monitor-extension[version'>=0.1.0,<1.0']

test/test_artifacts/v2/jupyter-activity-monitor-extension.test.Dockerfile

@@ -0,0 +1,8 @@
+ARG SAGEMAKER_DISTRIBUTION_IMAGE
+FROM $SAGEMAKER_DISTRIBUTION_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+RUN micromamba install pytest --freeze-installed --yes --channel conda-forge --name base
+RUN micromamba install pytest-jupyter --freeze-installed --yes --channel conda-forge --name base
+RUN SITE_PACKAGES=$(pip show jupyter-activity-monitor-extension | grep Location | awk '{print $2}') && \
+    cd "$SITE_PACKAGES/jupyter_activity_monitor_extension/tests/" && pytest -p pytest_jupyter.jupyter_server

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-activity-monitor-extension.test.Dockerfile", ["jupyter-activity-monitor-extension"]),
     ],
 )
 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-activity-monitor-extension.test.Dockerfile", ["jupyter-activity-monitor-extension"]),
         ("gpu-dependencies.test.Dockerfile", ["pytorch", "tensorflow"]),
     ],
 )