Merge pull request #768 from rewenila/bci_pytorch

Adding tests for AI pytorch container

Author: dcermak

bci_tester/data.py

@@ -1029,6 +1029,13 @@ def create_BCI(
     custom_entry_point="/bin/bash",
 )
 
+PYTORCH_CONTAINER = create_BCI(
+    build_tag=f"{SAC_CONTAINER_PREFIX}/pytorch:2-nvidia",
+    bci_type=ImageType.SAC_APPLICATION,
+    available_versions=["15.6-ai"],
+    custom_entry_point="/bin/bash",
+)
+
 STUNNEL_CONTAINER = create_BCI(
     build_tag=f"{APP_CONTAINER_PREFIX}/stunnel:5",
     bci_type=ImageType.APPLICATION,
@@ -1112,6 +1119,7 @@ def create_BCI(
     MINIMAL_CONTAINER,
     OLLAMA_CONTAINER,
     MILVUS_CONTAINER,
+    PYTORCH_CONTAINER,
     *POSTFIX_CONTAINERS,
     *TOMCAT_CONTAINERS,
     *POSTGRESQL_CONTAINERS,
@@ -1157,6 +1165,7 @@ def create_BCI(
             OLLAMA_CONTAINER,
             OPENWEBUI_CONTAINER,
             MILVUS_CONTAINER,
+            PYTORCH_CONTAINER,
         ]
         + BASE_FIPS_CONTAINERS
         + CONTAINER_389DS_CONTAINERS

pyproject.toml

@@ -141,6 +141,7 @@ markers = [
     'python_3.12',
     'python_3.13',
     'python_3.6',
+    'pytorch_2-nvidia',
     'registry_2.8',
     'ruby_2.5',
     'ruby_3.4',

tests/test_ai.py

@@ -9,8 +9,14 @@
 from bci_tester.data import MILVUS_CONTAINER
 from bci_tester.data import OLLAMA_CONTAINER
 from bci_tester.data import OPENWEBUI_CONTAINER
+from bci_tester.data import PYTORCH_CONTAINER
 
-CONTAINER_IMAGES = (OLLAMA_CONTAINER, OPENWEBUI_CONTAINER, MILVUS_CONTAINER)
+CONTAINER_IMAGES = (
+    OLLAMA_CONTAINER,
+    OPENWEBUI_CONTAINER,
+    MILVUS_CONTAINER,
+    PYTORCH_CONTAINER,
+)
 
 
 @pytest.mark.parametrize(
@@ -71,3 +77,18 @@ def test_milvus_health(container_per_test):
     )
     container_per_test.connection.check_output("etcd --version")
     container_per_test.connection.check_output("milvus")
+
+
+@pytest.mark.parametrize(
+    "container",
+    [PYTORCH_CONTAINER],
+    indirect=["container"],
+)
+def test_pytorch_health(container):
+    """Test the pytorch container."""
+
+    # chech pytorch Version
+    container.connection.check_output(
+        "python3.11 -c 'import torch; print(torch.__version__)'"
+    )
+    container.connection.check_output("git --version")

tests/test_metadata.py

@@ -83,6 +83,7 @@
 from bci_tester.data import POSTGRESQL_CONTAINERS
 from bci_tester.data import PROMETHEUS_CONTAINERS
 from bci_tester.data import PYTHON_CONTAINERS
+from bci_tester.data import PYTORCH_CONTAINER
 from bci_tester.data import RUBY_CONTAINERS
 from bci_tester.data import RUST_CONTAINERS
 from bci_tester.data import SAC_PYTHON_CONTAINERS
@@ -101,6 +102,10 @@
     reason="no supportlevel labels on openSUSE containers",
 )
 
+SKIP_IF_AI_MARK = pytest.mark.skipif(
+    OS_VERSION == "15.6-ai", reason="no supportlevel labels on AI containers"
+)
+
 
 def _get_container_label_prefix(
     container_name: str, container_type: ImageType
@@ -277,6 +282,7 @@ def _get_container_label_prefix(
         (OLLAMA_CONTAINER, "ollama", ImageType.SAC_APPLICATION),
         (OPENWEBUI_CONTAINER, "open-webui", ImageType.SAC_APPLICATION),
         (MILVUS_CONTAINER, "milvus", ImageType.SAC_APPLICATION),
+        (PYTORCH_CONTAINER, "pytorch", ImageType.SAC_APPLICATION),
     ]
     + [(STUNNEL_CONTAINER, "stunnel", ImageType.APPLICATION)]
     + [
@@ -516,6 +522,7 @@ def test_disturl_can_be_checked_out(
 
 
 @SKIP_IF_TW_MARK
+@SKIP_IF_AI_MARK
 @pytest.mark.parametrize(
     "container",
     [
@@ -552,6 +559,7 @@ def test_techpreview_label(container: ContainerData):
 
 
 @SKIP_IF_TW_MARK
+@SKIP_IF_AI_MARK
 @pytest.mark.parametrize(
     "container",
     list(ACC_CONTAINERS),
@@ -568,6 +576,7 @@ def test_acc_label(container: ContainerData):
 
 
 @SKIP_IF_TW_MARK
+@SKIP_IF_AI_MARK
 @pytest.mark.parametrize("container", L3_CONTAINERS, indirect=True)
 def test_l3_label(container: ContainerData):
     """Check that containers under L3 support have the label