feature: Add dependencies for amazon-sagemaker-sql extension

Author: kumarnzt

build_artifacts/v1/v1.6/v1.6.0/cpu.additional_packages_env.in

@@ -1,2 +1,4 @@
 conda-forge::jupyter_server_fileid
-conda-forge::jupyter_ydoc
+conda-forge::jupyter_ydoc
+conda-forge::amazon_sagemaker_sql_editor[version='>=0.1.0,<2']
+conda-forge::amazon-sagemaker-sql-magic[version='>=0.1.0,<2']

build_artifacts/v1/v1.6/v1.6.0/gpu.additional_packages_env.in

@@ -1,2 +1,4 @@
 conda-forge::jupyter_server_fileid
-conda-forge::jupyter_ydoc
+conda-forge::jupyter_ydoc
+conda-forge::amazon_sagemaker_sql_editor[version='>=0.1.0,<2']
+conda-forge::amazon-sagemaker-sql-magic[version='>=0.1.0,<2']

test/test_artifacts/v1/amazon-sagemaker-sql-magic.test.Dockerfile

@@ -0,0 +1,6 @@
+ARG COSMOS_IMAGE
+FROM $COSMOS_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+CMD ["python", "-c", "import amazon_sagemaker_sql_magic"]

test/test_artifacts/v1/amazon_sagemaker_sql_editor.test.Dockerfile

@@ -0,0 +1,6 @@
+ARG COSMOS_IMAGE
+FROM $COSMOS_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+CMD ["python", "-c", "import amazon_sagemaker_sql_editor"]

test/test_dockerfile_based_harness.py

@@ -40,7 +40,9 @@
     ("jupyterlab-git.test.Dockerfile", ['jupyterlab-git']),
     ("serve.test.Dockerfile", ['serve-langchain']),
     ("jupyter-server-fileid.test.Dockerfile", ['jupyter_server_fileid']),
-    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])])
+    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])],
+    ("amazon-sagemaker-sql-magic.test.Dockerfile", ['amazon-sagemaker-sql-magic']),
+    ("amazon_sagemaker_sql_editor.test.Dockerfile", ['amazon_sagemaker_sql_editor']))
 def test_dockerfiles_for_cpu(dockerfile_path: str, required_packages: List[str],
                              local_image_version: str, use_gpu: bool):
     _validate_docker_images(dockerfile_path, required_packages, local_image_version, use_gpu, 'cpu')
@@ -72,7 +74,9 @@ def test_dockerfiles_for_cpu(dockerfile_path: str, required_packages: List[str],
     ("jupyterlab-git.test.Dockerfile", ['jupyterlab-git']),
     ("serve.test.Dockerfile", ['serve-langchain']),
     ("jupyter-server-fileid.test.Dockerfile", ['jupyter_server_fileid']),
-    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])])
+    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])],
+    ("amazon-sagemaker-sql-magic.test.Dockerfile", ['amazon-sagemaker-sql-magic']),
+    ("amazon_sagemaker_sql_editor.test.Dockerfile", ['amazon_sagemaker_sql_editor']))
 def test_dockerfiles_for_gpu(dockerfile_path: str, required_packages: List[str],
                              local_image_version: str, use_gpu: bool):
     _validate_docker_images(dockerfile_path, required_packages, local_image_version, use_gpu, 'gpu')