Reinstates test suites

JasonWeill · Oct 12, 2023 · 0c81586 · 0c81586
1 parent a83f6a9
commit 0c81586
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py
@@ -19,15 +19,15 @@
 @pytest.mark.parametrize("dockerfile_path, required_packages", [
     ("keras.test.Dockerfile", ['keras']),
     ("autogluon.test.Dockerfile", ['autogluon']),
-    #    ("matplotlib.test.Dockerfile", ['matplotlib']),
-    #    ("scipy.test.Dockerfile", ['scipy']),
-    #    ("numpy.test.Dockerfile", ['numpy']),
-    #    ("boto3.test.Dockerfile", ['boto3']),
-    #    ("pandas.test.Dockerfile", ['pandas']),
-    #    ("sm-python-sdk.test.Dockerfile", ['sagemaker-python-sdk']),
-    #    ("pytorch.examples.Dockerfile", ['pytorch']),
-    #    ("tensorflow.examples.Dockerfile", ['tensorflow']),
-    #    ("jupyter-ai.test.Dockerfile", ['jupyter-ai']),
+    ("matplotlib.test.Dockerfile", ['matplotlib']),
+    ("scipy.test.Dockerfile", ['scipy']),
+    ("numpy.test.Dockerfile", ['numpy']),
+    ("boto3.test.Dockerfile", ['boto3']),
+    ("pandas.test.Dockerfile", ['pandas']),
+    ("sm-python-sdk.test.Dockerfile", ['sagemaker-python-sdk']),
+    ("pytorch.examples.Dockerfile", ['pytorch']),
+    ("tensorflow.examples.Dockerfile", ['tensorflow']),
+    ("jupyter-ai.test.Dockerfile", ['jupyter-ai']),
     ("jupyter-lsp.test.Dockerfile", ['jupyter-lsp']),
     ("notebook.test.Dockerfile", ['notebook'])])
 def test_dockerfiles_for_cpu(dockerfile_path: str, required_packages: List[str],
@@ -39,15 +39,15 @@ def test_dockerfiles_for_cpu(dockerfile_path: str, required_packages: List[str],
 @pytest.mark.parametrize("dockerfile_path, required_packages", [
     ("keras.test.Dockerfile", ['keras']),
     ("autogluon.test.Dockerfile", ['autogluon']),
-    #("matplotlib.test.Dockerfile", ['matplotlib']),
-    #("scipy.test.Dockerfile", ['scipy']),
-    #("numpy.test.Dockerfile", ['numpy']),
-    #("boto3.test.Dockerfile", ['boto3']),
-    #("pandas.test.Dockerfile", ['pandas']),
-    #("sm-python-sdk.test.Dockerfile", ['sagemaker-python-sdk']),
-    #("pytorch.examples.Dockerfile", ['pytorch']),
-    #("tensorflow.examples.Dockerfile", ['tensorflow']),
-    #("jupyter-ai.test.Dockerfile", ['jupyter-ai']),
+    ("matplotlib.test.Dockerfile", ['matplotlib']),
+    ("scipy.test.Dockerfile", ['scipy']),
+    ("numpy.test.Dockerfile", ['numpy']),
+    ("boto3.test.Dockerfile", ['boto3']),
+    ("pandas.test.Dockerfile", ['pandas']),
+    ("sm-python-sdk.test.Dockerfile", ['sagemaker-python-sdk']),
+    ("pytorch.examples.Dockerfile", ['pytorch']),
+    ("tensorflow.examples.Dockerfile", ['tensorflow']),
+    ("jupyter-ai.test.Dockerfile", ['jupyter-ai']),
     ("jupyter-lsp.test.Dockerfile", ['jupyter-lsp']),
     ("notebook.test.Dockerfile", ['notebook'])])
 def test_dockerfiles_for_gpu(dockerfile_path: str, required_packages: List[str],