notebooks: fix test failures in google_notebooks_instance (#14329) (#1025)

modular-magician · web-flow · commit 89a34c827579 · 2025-06-23T12:48:31.000-07:00
[upstream:fc7fba6ecba1a7395729f813bf37e22306ab132a]

Signed-off-by: Modular Magician &lt;magic-modules@google.com&gt;
diff --git a/notebook_instance_basic/main.tf b/notebook_instance_basic/main.tf
@@ -4,6 +4,6 @@ resource "google_notebooks_instance" "instance" {
   machine_type = "e2-medium"
   vm_image {
     project      = "deeplearning-platform-release"
-    image_family = "tf-latest-cpu"
+    image_family = "pytorch-latest-cu124"
   }
 }
diff --git a/notebook_instance_basic_gpu/main.tf b/notebook_instance_basic_gpu/main.tf
@@ -10,6 +10,6 @@ resource "google_notebooks_instance" "instance" {
   }
   vm_image {
     project      = "deeplearning-platform-release"
-    image_family = "tf-latest-gpu"
+    image_family = "pytorch-latest-cu124"
   }
 }
diff --git a/notebook_instance_basic_stopped/main.tf b/notebook_instance_basic_stopped/main.tf
@@ -4,7 +4,7 @@ resource "google_notebooks_instance" "instance" {
   machine_type = "e2-medium"
   vm_image {
     project      = "deeplearning-platform-release"
-    image_family = "tf-latest-cpu"
+    image_family = "pytorch-latest-cu124"
   }
   desired_state = "STOPPED"
 }

Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,6 @@ resource "google_notebooks_instance" "instance" {`
`4`	`4`	`machine_type = "e2-medium"`
`5`	`5`	`vm_image {`
`6`	`6`	`project = "deeplearning-platform-release"`
`7`		`- image_family = "tf-latest-cpu"`
	`7`	`+ image_family = "pytorch-latest-cu124"`
`8`	`8`	`}`
`9`	`9`	`}`
Original file line number	Diff line number	Diff line change
`@@ -10,6 +10,6 @@ resource "google_notebooks_instance" "instance" {`
`10`	`10`	`}`
`11`	`11`	`vm_image {`
`12`	`12`	`project = "deeplearning-platform-release"`
`13`		`- image_family = "tf-latest-gpu"`
	`13`	`+ image_family = "pytorch-latest-cu124"`
`14`	`14`	`}`
`15`	`15`	`}`
Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@ resource "google_notebooks_instance" "instance" {`
`4`	`4`	`machine_type = "e2-medium"`
`5`	`5`	`vm_image {`
`6`	`6`	`project = "deeplearning-platform-release"`
`7`		`- image_family = "tf-latest-cpu"`
	`7`	`+ image_family = "pytorch-latest-cu124"`
`8`	`8`	`}`
`9`	`9`	`desired_state = "STOPPED"`
`10`	`10`	`}`