Skip to content

Commit

Permalink
Add jupyter-idle extension
Browse files Browse the repository at this point in the history
- Added the package to the {cpu/gpu}.additional_packages_env.in files
- Added import test for the jupyter-idle extension package
- Built the images successfully and tested the extension on jupyter lab
  spun up on the image. Results were as expected.
- An OSS review and Appsec review has been copleted for this package
- pypi : https://pypi.org/project/jupyter-idle/0.1.1/
- conda-forge : https://anaconda.org/conda-forge/jupyter-idle
  • Loading branch information
rahrad123 committed Jul 8, 2024
1 parent 3f1f959 commit 919fce8
Show file tree
Hide file tree
Showing 4 changed files with 10 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -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']
Original file line number Diff line number Diff line change
@@ -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']
6 changes: 6 additions & 0 deletions test/test_artifacts/v2/jupyter-idle-ext.test.Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
ARG SAGEMAKER_DISTRIBUTION_IMAGE
FROM $SAGEMAKER_DISTRIBUTION_IMAGE

ARG MAMBA_DOCKERFILE_ACTIVATE=1

CMD ["python", "-c", "import jupyter_idle"]
2 changes: 2 additions & 0 deletions test/test_dockerfile_based_harness.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit 919fce8

Please sign in to comment.