From ce73fe48f7246a1d6b7da2bec7739914c0cea669 Mon Sep 17 00:00:00 2001
From: Yiying Zhang <yiyingzhang@google.com>
Date: Tue, 13 Aug 2024 15:44:50 -0700
Subject: [PATCH] bump jupyter/tensorflow-notebook to python-3.11

---
 applications/jupyter/main.tf                                 | 2 +-
 modules/jupyter/jupyter_image/notebook_image/Dockerfile      | 2 +-
 modules/jupyter/jupyter_image/notebook_image/cloudbuild.yaml | 2 +-
 modules/jupyter/variables.tf                                 | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/applications/jupyter/main.tf b/applications/jupyter/main.tf
index 3998a5255..663de068c 100644
--- a/applications/jupyter/main.tf
+++ b/applications/jupyter/main.tf
@@ -148,7 +148,7 @@ module "jupyterhub" {
   gcs_bucket                        = var.gcs_bucket
   autopilot_cluster                 = local.enable_autopilot
   notebook_image                    = "jupyter/tensorflow-notebook"
-  notebook_image_tag                = "python-3.10"
+  notebook_image_tag                = "python-3.11"
 
   # IAP Auth parameters
   add_auth                 = var.add_auth
diff --git a/modules/jupyter/jupyter_image/notebook_image/Dockerfile b/modules/jupyter/jupyter_image/notebook_image/Dockerfile
index 302691aa5..5517ed467 100644
--- a/modules/jupyter/jupyter_image/notebook_image/Dockerfile
+++ b/modules/jupyter/jupyter_image/notebook_image/Dockerfile
@@ -1,3 +1,3 @@
-FROM jupyter/tensorflow-notebook:python-3.10
+FROM jupyter/tensorflow-notebook:python-3.11
 COPY requirements.txt ./requirements.txt
 RUN pip install --no-cache-dir -r ./requirements.txt
diff --git a/modules/jupyter/jupyter_image/notebook_image/cloudbuild.yaml b/modules/jupyter/jupyter_image/notebook_image/cloudbuild.yaml
index d1a4acfa2..6aacfac71 100644
--- a/modules/jupyter/jupyter_image/notebook_image/cloudbuild.yaml
+++ b/modules/jupyter/jupyter_image/notebook_image/cloudbuild.yaml
@@ -15,7 +15,7 @@
 # to build, run `gcloud builds submit --config cloudbuild.yaml .` in directory
 steps:
 - name: 'gcr.io/cloud-builders/docker'
-  args: [ 'pull', 'docker.io/jupyter/tensorflow-notebook:python-3.10' ]
+  args: [ 'pull', 'docker.io/jupyter/tensorflow-notebook:python-3.11' ]
 - name: 'gcr.io/cloud-builders/docker'
   args: [ 'build', '-t', '<Artiact registry repo>/<image name>', '.' ]
 images:
diff --git a/modules/jupyter/variables.tf b/modules/jupyter/variables.tf
index 845838616..b8f3e4a2b 100644
--- a/modules/jupyter/variables.tf
+++ b/modules/jupyter/variables.tf
@@ -26,7 +26,7 @@ variable "notebook_image" {
 variable "notebook_image_tag" {
   type        = string
   description = "Jupyter notebook image tag"
-  default     = "python-3.10"
+  default     = "python-3.11"
 }
 
 variable "members_allowlist" {