diff --git a/src/main/java/io/odh/test/framework/manager/resources/NotebookResource.java b/src/main/java/io/odh/test/framework/manager/resources/NotebookResource.java index 8c2cd40e..2f65b4b0 100644 --- a/src/main/java/io/odh/test/framework/manager/resources/NotebookResource.java +++ b/src/main/java/io/odh/test/framework/manager/resources/NotebookResource.java @@ -26,18 +26,18 @@ public class NotebookResource implements ResourceType { private static final String REGISTRY_PATH = "image-registry.openshift-image-registry.svc:5000"; - public static final String PYTORCH_IMAGE = "pytorch"; - public static final String PYTORCH_2023_2_TAG = "2023.2"; + public static final String JUPYTER_MINIMAL_IMAGE = "jupyter-minimal-notebook"; + public static final String JUPYTER_MINIMAL_2023_2_TAG = "2023.2"; private static final Map ODH_IMAGES_MAP; static { - ODH_IMAGES_MAP = Map.of(PYTORCH_IMAGE, "jupyter-pytorch-notebook"); + ODH_IMAGES_MAP = Map.of(JUPYTER_MINIMAL_IMAGE, "jupyter-minimal-notebook"); } private static final Map RHOAI_IMAGES_MAP; static { - RHOAI_IMAGES_MAP = Map.of(PYTORCH_IMAGE, "pytorch"); + RHOAI_IMAGES_MAP = Map.of(JUPYTER_MINIMAL_IMAGE, "s2i-minimal-notebook"); } private static final String NOTEBOOK_TEMPLATE_PATH = "notebook.yaml"; @Override diff --git a/src/test/java/io/odh/test/e2e/standard/NotebookST.java b/src/test/java/io/odh/test/e2e/standard/NotebookST.java index 90824486..376bb7ee 100644 --- a/src/test/java/io/odh/test/e2e/standard/NotebookST.java +++ b/src/test/java/io/odh/test/e2e/standard/NotebookST.java @@ -111,7 +111,7 @@ void testCreateSimpleNotebook() throws IOException { .build(); ResourceManager.getInstance().createResourceWithoutWait(pvc); - String notebookImage = NotebookResource.getNotebookImage(NotebookResource.PYTORCH_IMAGE, NotebookResource.PYTORCH_2023_2_TAG); + String notebookImage = NotebookResource.getNotebookImage(NotebookResource.JUPYTER_MINIMAL_IMAGE, NotebookResource.JUPYTER_MINIMAL_2023_2_TAG); Notebook notebook = new NotebookBuilder(NotebookResource.loadDefaultNotebook(NTB_NAMESPACE, NTB_NAME, notebookImage)).build(); ResourceManager.getInstance().createResourceWithoutWait(notebook); diff --git a/src/test/java/io/odh/test/e2e/upgrade/UpgradeAbstract.java b/src/test/java/io/odh/test/e2e/upgrade/UpgradeAbstract.java index 0f9e8e57..57e051c0 100644 --- a/src/test/java/io/odh/test/e2e/upgrade/UpgradeAbstract.java +++ b/src/test/java/io/odh/test/e2e/upgrade/UpgradeAbstract.java @@ -114,7 +114,7 @@ public void deployNotebook(String namespace, String name) throws IOException { .build(); ResourceManager.getInstance().createResourceWithoutWait(pvc); - String notebookImage = NotebookResource.getNotebookImage(NotebookResource.PYTORCH_IMAGE, NotebookResource.PYTORCH_2023_2_TAG); + String notebookImage = NotebookResource.getNotebookImage(NotebookResource.JUPYTER_MINIMAL_IMAGE, NotebookResource.JUPYTER_MINIMAL_2023_2_TAG); Notebook notebook = new NotebookBuilder(NotebookResource.loadDefaultNotebook(namespace, name, notebookImage)).build(); if (!Environment.PRODUCT.equals(Environment.PRODUCT_DEFAULT)) { notebook = new NotebookBuilder(NotebookResource.loadDefaultNotebook(namespace, name, notebookImage))