From 19e0f2135907b5e7f952132725c010b260716fd6 Mon Sep 17 00:00:00 2001
From: Jacob Tomlinson <jtomlinson@nvidia.com>
Date: Wed, 11 Oct 2023 17:18:11 +0100
Subject: [PATCH] Update Kubernetes docs to use notebooks container

---
 source/conf.py                 | 2 ++
 source/platforms/kubernetes.md | 4 ++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/source/conf.py b/source/conf.py
index 0fff7535..f4388ca7 100644
--- a/source/conf.py
+++ b/source/conf.py
@@ -28,12 +28,14 @@
     "stable": {
         "rapids_version": stable_version,
         "rapids_container": f"nvcr.io/nvidia/rapidsai/base:{stable_version}-cuda11.8-py3.10",
+        "rapids_notebooks_container": f"nvcr.io/nvidia/rapidsai/notebooks:{stable_version}-cuda11.8-py3.10",
         "rapids_conda_channels": "-c rapidsai -c conda-forge -c nvidia",
         "rapids_conda_packages": f"rapids={stable_version} python=3.10 cudatoolkit=11.8",
     },
     "nightly": {
         "rapids_version": f"{nightly_version}-nightly",
         "rapids_container": f"rapidsai/base:{nightly_version + 'a'}-cuda11.8-py3.10",
+        "rapids_notebooks_container": f"rapidsai/notebooks:{nightly_version + 'a'}-cuda11.8-py3.10",
         "rapids_conda_channels": "-c rapidsai-nightly -c conda-forge -c nvidia",
         "rapids_conda_packages": f"rapids={nightly_version} python=3.10 cudatoolkit=11.8",
     },
diff --git a/source/platforms/kubernetes.md b/source/platforms/kubernetes.md
index 5a40fa07..69dd78ab 100644
--- a/source/platforms/kubernetes.md
+++ b/source/platforms/kubernetes.md
@@ -39,7 +39,7 @@ spec:
     fsGroup: 0
   containers:
     - name: rapids-notebook
-      image: "{{ rapids_container }}"
+      image: "{{ rapids_notebooks_container }}"
       resources:
         limits:
           nvidia.com/gpu: 1
@@ -233,7 +233,7 @@ spec:
     fsGroup: 0
   containers:
     - name: rapids-notebook
-      image: {{ rapids_container }}
+      image: {{ rapids_notebooks_container }}
       resources:
         limits:
           nvidia.com/gpu: 1